#222: Fix pyop2 cache clear 2

mesh-adaptation · Oct 14, 2024 · 1a83733 · 1a83733
1 parent 73e38f4
commit 1a83733
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/test/conftest.py b/test/conftest.py
@@ -23,6 +23,7 @@ def pytest_runtest_teardown(item, nextitem):
     """
     from firedrake.tsfc_interface import clear_cache
     from pyop2.caching import clear_memory_cache
+    from pyop2.mpi import COMM_WORLD
 
     clear_cache()
-    clear_memory_cache()
+    clear_memory_cache(COMM_WORLD)
diff --git a/test_adjoint/conftest.py b/test_adjoint/conftest.py
@@ -137,9 +137,10 @@ def pytest_runtest_teardown(item, nextitem):
     """
     from firedrake.tsfc_interface import clear_cache
     from pyop2.caching import clear_memory_cache
+    from pyop2.mpi import COMM_WORLD
 
     clear_cache()
-    clear_memory_cache()
+    clear_memory_cache(COMM_WORLD)
 
 
 @pytest.fixture(autouse=True)