projects
/
pytorch.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
| inline |
side by side
(parent:
ab248b1
)
Oups.
author
Francois Fleuret
<francois@fleuret.org>
Thu, 3 Sep 2020 04:41:21 +0000
(06:41 +0200)
committer
Francois Fleuret
<francois@fleuret.org>
Thu, 3 Sep 2020 04:41:21 +0000
(06:41 +0200)
speed.py
patch
|
blob
|
history
diff --git
a/speed.py
b/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