Ignore cleanup errors in cache teardown. (#3020)

To avoid PermissionError on Windows where certain .pyd files are locked by current Python process and cannot be deleted. Fixes #3019.
intel · Dec 17, 2024 · 650428b · 650428b
1 parent f1a893a
commit 650428b
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 12 deletions.
diff --git a/python/test/regression/conftest.py b/python/test/regression/conftest.py
@@ -14,9 +14,13 @@ def device(request):
 
 @pytest.fixture
 def fresh_triton_cache():
-    with tempfile.TemporaryDirectory() as tmpdir:
-        try:
-            os.environ["TRITON_CACHE_DIR"] = tmpdir
-            yield tmpdir
-        finally:
-            os.environ.pop("TRITON_CACHE_DIR", None)
+    try:
+        with tempfile.TemporaryDirectory() as tmpdir:
+            try:
+                os.environ["TRITON_CACHE_DIR"] = tmpdir
+                yield tmpdir
+            finally:
+                os.environ.pop("TRITON_CACHE_DIR", None)
+    except OSError:
+        # Ignore errors, such as PermissionError, on Windows
+        pass
diff --git a/python/test/unit/conftest.py b/python/test/unit/conftest.py
@@ -14,9 +14,13 @@ def device(request):
 
 @pytest.fixture
 def fresh_triton_cache():
-    with tempfile.TemporaryDirectory() as tmpdir:
-        try:
-            os.environ["TRITON_CACHE_DIR"] = tmpdir
-            yield tmpdir
-        finally:
-            os.environ.pop("TRITON_CACHE_DIR", None)
+    try:
+        with tempfile.TemporaryDirectory() as tmpdir:
+            try:
+                os.environ["TRITON_CACHE_DIR"] = tmpdir
+                yield tmpdir
+            finally:
+                os.environ.pop("TRITON_CACHE_DIR", None)
+    except OSError:
+        # Ignore errors, such as PermissionError, on Windows
+        pass