also address scan kernel cache

inducer · Apr 23, 2024 · 980c3db · 980c3db
1 parent 226a52e
commit 980c3db
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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(