Skip to content

Commit 267d735

Browse files
committed
fix: NoneType object is not subscriptable for setup_dict and cleanup_dict
1 parent a4104c0 commit 267d735

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

kernel_tuner/core.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -616,7 +616,7 @@ def compile_and_benchmark(self, kernel_source, gpu_args, params, kernel_options,
616616
start_benchmark = time.perf_counter()
617617

618618
# Run code before benchmarking when using the compiler backend
619-
if "setup_dict" in kernel_options and isinstance(self.dev, CompilerFunctions):
619+
if kernel_options["setup_dict"] and isinstance(self.dev, CompilerFunctions):
620620
setup_dict = kernel_options["setup_dict"]
621621
args_len = setup_dict["args_len"]
622622
kernel_name = kernel_options["kernel_name"]
@@ -632,7 +632,7 @@ def compile_and_benchmark(self, kernel_source, gpu_args, params, kernel_options,
632632
last_benchmark_time = 1000 * (time.perf_counter() - start_benchmark)
633633

634634
# Run code after benchmarking when using the compiler backend
635-
if "cleanup_dict" in kernel_options and isinstance(self.dev, CompilerFunctions):
635+
if kernel_options["cleanup_dict"] and isinstance(self.dev, CompilerFunctions):
636636
cleanup_dict = kernel_options["cleanup_dict"]
637637
args_len = cleanup_dict["args_len"]
638638
kernel_name = kernel_options["kernel_name"]

0 commit comments

Comments
 (0)