diff --git a/loopy/codegen/__init__.py b/loopy/codegen/__init__.py index cf32c6885aa963b14145241d9f23eda68ef7228d..5977ade1191ed93d62c8f6445872262c54d47b02 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 7dbc8cb686722740821ee890ba0b8fc244d39f6e..f6fbdb41acc783b2339c86bcf554162c169d24f2 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 ad708101b6d9ffd04eac1ec5ff39b95e1b8fc2b3..8d45bf723f6e87c7a940f388ae7f1b0a16d93aac 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 ca4256b0dfbd25f1964c888cda7c408cbf63b461..789e3b373cbf47a8768429a94f39baac09bd9ea0 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 17f68968f42bef04725d67bafc09f05a458d1e5a..dc1ef4cbf7f3973a46ed611f4f64277b5e8bb592 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",