From e8d861a56f509fa352b6f7bf3b099e42233044e2 Mon Sep 17 00:00:00 2001 From: Yukio Siraichi Date: Fri, 8 Dec 2023 17:06:13 -0300 Subject: [PATCH] Remove branch on undefined variable. --- benchmarks/experiment_runner.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/benchmarks/experiment_runner.py b/benchmarks/experiment_runner.py index 124eebed0bd..05c5861ce4e 100644 --- a/benchmarks/experiment_runner.py +++ b/benchmarks/experiment_runner.py @@ -276,8 +276,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