Skip to content

Commit 00c215a

Browse files
committed
include default ncu report
1 parent f31e4bb commit 00c215a

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

src/libkernelbot/run_eval.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -417,9 +417,14 @@ def profile_program_ncu(
417417
run_result = run_program(call, seed=seed, timeout=timeout, multi_gpu=multi_gpu, extra_env={
418418
"POPCORN_NCU": "1"
419419
})
420-
421420
profile_result = None
422421

422+
try:
423+
report = subprocess.check_output(["ncu", "--import", f"{str(output_dir / 'profile.ncu-rep')}",], text=True)
424+
run_result.result["benchmark.0.report"] = base64.b64encode(report.encode("utf-8")).decode("utf-8")
425+
except subprocess.CalledProcessError:
426+
pass
427+
423428
if run_result.success:
424429
profile_result = ProfileResult(
425430
profiler='Nsight-Compute',

0 commit comments

Comments
 (0)