We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ca860b3 commit c21717fCopy full SHA for c21717f
graph_net/torch/post_extract_process.py
@@ -74,6 +74,6 @@ def count_kernels(model, sample_inputs) -> int:
74
75
total_count = 0
76
for e in events:
77
- if e.key == "cuLaunchKernel" or e.key == "cudalaunchKernel":
+ if e.key == "cuLaunchKernel" or e.key == "cudaLaunchKernel":
78
total_count += e.count
79
return total_count
0 commit comments