Oups.
authorFrancois Fleuret <francois@fleuret.org>
Thu, 3 Sep 2020 04:41:21 +0000 (06:41 +0200)
committerFrancois Fleuret <francois@fleuret.org>
Thu, 3 Sep 2020 04:41:21 +0000 (06:41 +0200)
speed.py

index 10a008f..e4add26 100755 (executable)
--- a/speed.py
+++ b/speed.py
@@ -14,11 +14,11 @@ d1, d2, d3 = 2048, 2048, 2048
 
 a, b = torch.rand(d1, d2).to(device), torch.rand(d2, d3).to(device)
 
-sync
+sync()
 start_time = time.perf_counter()
 for k in range(nb_runs):
     c = a @ b
-sync
+sync()
 duration = time.perf_counter() - start_time
 
 nb_flop = float(nb_runs * d1 * d2 * d3 * 2) # 1 multiply-and-add is 2 ops