Skip to content

GitLab

Commit 0d9c540f authored by Rick Ho's avatar Rick Ho
Browse files

fix timing bug

parent 8cff6ad7
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
pytorch/cuda/moe_test.py
View file @ 0d9c540f
...@@ -26,7 +26,7 @@ def perf(): ...@@ -26,7 +26,7 @@ def perf():
te = time.time() te = time.time()
tott += te - ts tott += te - ts
gflops = 2e-9 * n_runs * in_feat * out_feat * batch_size gflops = 2e-9 * n_runs * in_feat * out_feat * batch_size / tott
print('Mean time {:.3f} ms, {:.3f} GFLOPs'.format(tott * 1e3 / n_runs, gflops)) print('Mean time {:.3f} ms, {:.3f} GFLOPs'.format(tott * 1e3 / n_runs, gflops))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment