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.

Author: pbchekin

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

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