Skip to content

Commit c21717f

Browse files
committed
modify the way of counting kernels used
1 parent ca860b3 commit c21717f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

graph_net/torch/post_extract_process.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,6 +74,6 @@ def count_kernels(model, sample_inputs) -> int:
7474

7575
total_count = 0
7676
for e in events:
77-
if e.key == "cuLaunchKernel" or e.key == "cudalaunchKernel":
77+
if e.key == "cuLaunchKernel" or e.key == "cudaLaunchKernel":
7878
total_count += e.count
7979
return total_count

0 commit comments

Comments
 (0)