Skip to content

Commit 09827b4

Browse files
ensure device memory is released
1 parent a97ff8b commit 09827b4

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

kernel_tuner/interface.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -368,6 +368,8 @@ def tune_kernel(kernel_name, kernel_string, problem_size, arguments,
368368
else:
369369
print("no results to report")
370370

371+
del runner.dev
372+
371373
return results, env
372374

373375

0 commit comments

Comments
 (0)