[DRIVER] Now giving the option to use system ptxas through environment variable (#123)

2021-06-11 13:48:11 -04:00
parent 80c86ecf4a
commit b7b05a560e
3 changed files with 40 additions and 72 deletions
--- a/python/tutorials/02-fused-softmax.py
+++ b/python/tutorials/02-fused-softmax.py
@@ -153,7 +153,7 @@ def benchmark(M, N, provider):
    return gbps(ms), gbps(max_ms), gbps(min_ms)


-benchmark.run(show_plots=True)
+benchmark.run(show_plots=True, print_data=True)

 # %%
 # In the above plot, we can see that: