fbcode/dietgpu/benchmark.py

Reviewed By: JuanBesa

Differential Revision: D90011010

fbshipit-source-id: 924654f978dbcb04f970bf4a6743ecfbcc30ca8a

Author: generatedunixname537391475639613

dietgpu/benchmark.py

@@ -98,7 +98,7 @@ def get_any_comp_timings(ts, num_runs=3):
     comp_size = 0
 
     # ignore first run timings
-    for i in range(1 + num_runs):
+    for _ in range(1 + num_runs):
         start = torch.cuda.Event(enable_timing=True)
         end = torch.cuda.Event(enable_timing=True)
 
@@ -168,7 +168,7 @@ def get_any_comp_timings(ts, num_runs=3):
 
     # Batched
     ts = []
-    for i in range(128):
+    for _ in range(128):
         ts.append(torch.normal(0, 1.0, [512 * 1024], dtype=dt, device=dev))
 
     c, dc, total_size, comp_size = get_float_comp_timings(ts)
@@ -206,7 +206,7 @@ def get_any_comp_timings(ts, num_runs=3):
 
     # Batched
     ts = []
-    for i in range(128):
+    for _ in range(128):
         ts.append(torch.normal(0, 1.0, [512 * 1024], dtype=dt, device=dev))
 
     c, dc, total_size, comp_size = get_any_comp_timings(ts)