Skip to content

Commit 59de19a

Browse files
committed
fix
Signed-off-by: Anatoly Myachev <[email protected]>
1 parent 1254163 commit 59de19a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

benchmarks/triton_kernels_benchmark/benchmark_testing.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -223,7 +223,7 @@ def do_bench_upstream_pytorch_profiler(fn, n_warmup=25, n_repeat=100, grad_to_no
223223
# profiling_func_filter = filter(lambda x: x.name.startswith("__profile_kernel_of_func"), function_events)
224224

225225
# Make the time to the milliseconds.
226-
times = torch.tensor([sum(map(lambda elem: elem.self_device_time_total), f) * 1e-3 for f in all_functions],
226+
times = torch.tensor([sum(map(lambda elem: elem.self_device_time_total, f)) * 1e-3 for f in all_functions],
227227
dtype=torch.float)
228228
return _summarize_statistics(times, quantiles, return_mode)
229229

0 commit comments

Comments
 (0)