From 636ee6c94874289d2ed0f3d850d5450748f65f9b Mon Sep 17 00:00:00 2001
From: Matthias Diener <mdiener@illinois.edu>
Date: Fri, 7 Jun 2024 17:20:28 -0500
Subject: [PATCH] unset safe_sync=False for caches

---
 loopy/codegen/__init__.py  | 3 ++-
 loopy/schedule/__init__.py | 3 ++-
 loopy/target/execution.py  | 6 ++++--
 loopy/tools.py             | 3 ++-
 setup.py                   | 2 +-
 5 files changed, 11 insertions(+), 6 deletions(-)

diff --git a/loopy/codegen/__init__.py b/loopy/codegen/__init__.py
index cf32c6885..5977ade11 100644
--- a/loopy/codegen/__init__.py
+++ b/loopy/codegen/__init__.py
@@ -317,7 +317,8 @@ code_gen_cache: WriteOncePersistentDict[
     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
index 7dbc8cb68..f6fbdb41a 100644
--- a/loopy/schedule/__init__.py
+++ b/loopy/schedule/__init__.py
@@ -2204,7 +2204,8 @@ schedule_cache: WriteOncePersistentDict[
         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
index ad708101b..8d45bf723 100644
--- a/loopy/target/execution.py
+++ b/loopy/target/execution.py
@@ -726,7 +726,8 @@ typed_and_scheduled_cache: WriteOncePersistentDict[
     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 @@ invoker_cache: WriteOncePersistentDict[
     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
index ca4256b0d..789e3b373 100644
--- 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
index 17f68968f..dc1ef4cbf 100644
--- a/setup.py
+++ b/setup.py
@@ -84,7 +84,7 @@ setup(name="loopy",
 
       python_requires="~=3.8",
       install_requires=[
-          "pytools>=2024.1.2",
+          "pytools>=2024.1.5",
           "pymbolic>=2022.1",
           "genpy>=2016.1.2",
 
-- 
GitLab