From 1be001b656ce2cb9e89d4741479dda0f1ea26096 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Mon, 13 Jun 2022 16:11:52 -0500
Subject: [PATCH] Obey PYOPENCL_NO_CACHE in scan generation
---
pyopencl/scan.py | 22 ++++++++++++----------
1 file changed, 12 insertions(+), 10 deletions(-)
diff --git a/pyopencl/scan.py b/pyopencl/scan.py
index 414cbd5d..66a46c0a 100644
--- a/pyopencl/scan.py
+++ b/pyopencl/scan.py
@@ -1152,17 +1152,19 @@ class GenericScanKernel(_GenericScanKernelBase):
from_cache = False
- try:
- result = generic_scan_kernel_cache[cache_key]
- from_cache = True
- logger.debug(
+ import os
+ if not os.environ.get("PYOPENCL_NO_CACHE"):
+ try:
+ result = generic_scan_kernel_cache[cache_key]
+ from_cache = True
+ logger.debug(
"cache hit for generated scan kernel '%s'" % self.name_prefix)
- (
- self.first_level_scan_gen_info,
- self.second_level_scan_gen_info,
- self.final_update_gen_info) = result
- except KeyError:
- pass
+ (
+ self.first_level_scan_gen_info,
+ self.second_level_scan_gen_info,
+ self.final_update_gen_info) = result
+ except KeyError:
+ pass
if not from_cache:
logger.debug(
--
GitLab