From: Francois Fleuret Date: Thu, 3 Sep 2020 04:41:21 +0000 (+0200) Subject: Oups. X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=31ffed8a581c04094e8d2727c88de6e1d2c07a65;p=pytorch.git Oups. --- 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