From 31ffed8a581c04094e8d2727c88de6e1d2c07a65 Mon Sep 17 00:00:00 2001 From: Francois Fleuret Date: Thu, 3 Sep 2020 06:41:21 +0200 Subject: [PATCH] Oups. --- speed.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/speed.py b/speed.py index 10a008f..e4add26 100755 --- 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 -- 2.39.5