Remove branch on undefined variable. (#6072)

pytorch · Dec 15, 2023 · c3e341c · c3e341c
1 parent 4103eb6
commit c3e341c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/benchmarks/experiment_runner.py b/benchmarks/experiment_runner.py
@@ -262,8 +262,7 @@ def collect_profile_to_metrics(self, prof, metrics):
       if evt.device_type == DeviceType.CPU:
         # in legacy profiler, kernel info is stored in cpu events
         if evt.is_legacy:
-          if not use_device:
-            total_cuda_time += evt.self_cuda_time_total
+          total_cuda_time += evt.self_cuda_time_total
       elif evt.device_type == DeviceType.CUDA:
         # in kineto profiler, there're events with the correct device type (e.g. CUDA)
         total_cuda_time += evt.self_cuda_time_total