Skip to content

Commit 3adb49b

Browse files
committed
limit number of kernels to be profiled
1 parent d3518a1 commit 3adb49b

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/libkernelbot/run_eval.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -451,6 +451,7 @@ def profile_program_ncu(
451451
"--nvtx",
452452
"--nvtx-include", "custom_kernel/",
453453
"--import-source", "1",
454+
"-c", "10",
454455
"-o", f"{str(output_dir / 'profile.ncu-rep')}",
455456
"--",
456457
] + call

0 commit comments

Comments
 (0)