support aborting on cache miss

inducer · Feb 28, 2024 · 8fd54b7 · 8fd54b7
1 parent 5f098fa
commit 8fd54b7
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/loopy/__init__.py b/loopy/__init__.py
@@ -427,6 +427,9 @@ def register_symbol_manglers(kernel, manglers):
     and
     "CG_NO_CACHE" not in os.environ)
 
+from pytools import strtobool
+ABORT_ON_CACHE_MISS = strtobool(os.environ.get("LOOPY_ABORT_ON_CACHE_MISS", "False"))
+
 
 def set_caching_enabled(flag):
     """Set whether :mod:`loopy` is allowed to use disk caching for its various

diff --git a/loopy/codegen/__init__.py b/loopy/codegen/__init__.py
@@ -573,7 +573,7 @@ def generate_code_v2(program):
 
     # {{{ cache retrieval
 
-    from loopy import CACHING_ENABLED
+    from loopy import CACHING_ENABLED, ABORT_ON_CACHE_MISS
 
     if CACHING_ENABLED:
         input_program = program
@@ -583,6 +583,9 @@ def generate_code_v2(program):
                           " code generation cache hit")
             return result
         except KeyError:
+            if ABORT_ON_CACHE_MISS:
+                raise
+
             logger.debug(f"TranslationUnit with entrypoints {program.entrypoints}:"
                           " code generation cache miss")