unset safe_sync=False for caches

inducer · Jun 8, 2024 · 636ee6c · 636ee6c
1 parent 37d66a5
commit 636ee6c
Show file tree

Hide file tree

Showing 5 changed files with 11 additions and 6 deletions.
diff --git a/loopy/codegen/__init__.py b/loopy/codegen/__init__.py
@@ -317,7 +317,8 @@ def ast_builder(self):
     CodeGenerationResult
 ] = WriteOncePersistentDict(
          "loopy-code-gen-cache-v3-"+DATA_MODEL_VERSION,
-         key_builder=LoopyKeyBuilder())
+         key_builder=LoopyKeyBuilder(),
+         safe_sync=False)
 
 
 caches.append(code_gen_cache)

diff --git a/loopy/schedule/__init__.py b/loopy/schedule/__init__.py
@@ -2204,7 +2204,8 @@ def print_longest_dead_end():
         LoopKernel
 ] = WriteOncePersistentDict(
         "loopy-schedule-cache-v4-"+DATA_MODEL_VERSION,
-        key_builder=LoopyKeyBuilder())
+        key_builder=LoopyKeyBuilder(),
+        safe_sync=False)
 
 
 caches.append(schedule_cache)

diff --git a/loopy/target/execution.py b/loopy/target/execution.py
@@ -726,7 +726,8 @@ def __call__(self, program, entrypoint, codegen_result):
     TranslationUnit
 ] = WriteOncePersistentDict(
         "loopy-typed-and-scheduled-cache-v1-"+DATA_MODEL_VERSION,
-        key_builder=LoopyKeyBuilder())
+        key_builder=LoopyKeyBuilder(),
+        safe_sync=False)
 
 
 caches.append(typed_and_scheduled_cache)
@@ -737,7 +738,8 @@ def __call__(self, program, entrypoint, codegen_result):
     str
 ] = WriteOncePersistentDict(
         "loopy-invoker-cache-v10-"+DATA_MODEL_VERSION,
-        key_builder=LoopyKeyBuilder())
+        key_builder=LoopyKeyBuilder(),
+        safe_sync=False)
 
 
 caches.append(invoker_cache)

diff --git a/loopy/tools.py b/loopy/tools.py
@@ -916,7 +916,8 @@ def memoize_on_disk(func, key_builder_t=LoopyKeyBuilder):
             f"{func.__name__}-"
             f"{key_builder_t.__qualname__}.{key_builder_t.__name__}"
             f"-v0-{DATA_MODEL_VERSION}"),
-        key_builder=key_builder_t())
+        key_builder=key_builder_t(),
+        safe_sync=False)
 
     caches.append(transform_cache)
 

diff --git a/setup.py b/setup.py
@@ -84,7 +84,7 @@ def write_git_revision(package_name):
 
       python_requires="~=3.8",
       install_requires=[
-          "pytools>=2024.1.2",
+          "pytools>=2024.1.5",
           "pymbolic>=2022.1",
           "genpy>=2016.1.2",