From 980c3dbe4e9b7f4cacc65a652c1e3df0aa2eeb5a Mon Sep 17 00:00:00 2001 From: Matthias Diener Date: Mon, 22 Apr 2024 16:01:38 -0500 Subject: [PATCH] also address scan kernel cache --- pyopencl/scan.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/pyopencl/scan.py b/pyopencl/scan.py index 4c1fc8d19..320b6e137 100644 --- a/pyopencl/scan.py +++ b/pyopencl/scan.py @@ -1141,7 +1141,8 @@ def finish_setup(self) -> None: pass -generic_scan_kernel_cache = WriteOncePersistentDict( +if not cl._PYOPENCL_NO_CACHE: + generic_scan_kernel_cache = WriteOncePersistentDict( "pyopencl-generated-scan-kernel-cache-v1", key_builder=_NumpyTypesKeyBuilder(), in_mem_cache_size=0) @@ -1199,7 +1200,8 @@ def finish_setup(self) -> None: self.second_level_scan_gen_info, self.final_update_gen_info) - generic_scan_kernel_cache.store_if_not_present(cache_key, result) + if not cl._PYOPENCL_NO_CACHE: + generic_scan_kernel_cache.store_if_not_present(cache_key, result) # Build the kernels. self.first_level_scan_info = self.first_level_scan_gen_info.build(