DEBUG: revert cleaning cache changes on Windows

Signed-off-by: Anatoly Myachev <[email protected]>

Author: anmyachev

python/test/regression/conftest.py

@@ -14,13 +14,9 @@ def device(request):
 
 @pytest.fixture
 def fresh_triton_cache():
-    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
+    with tempfile.TemporaryDirectory() as tmpdir:
+        try:
+            os.environ["TRITON_CACHE_DIR"] = tmpdir
+            yield tmpdir
+        finally:
+            os.environ.pop("TRITON_CACHE_DIR", None)

python/test/unit/conftest.py

@@ -14,16 +14,12 @@ def device(request):
 
 @pytest.fixture
 def fresh_triton_cache():
-    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
+    with tempfile.TemporaryDirectory() as tmpdir:
+        try:
+            os.environ["TRITON_CACHE_DIR"] = tmpdir
+            yield tmpdir
+        finally:
+            os.environ.pop("TRITON_CACHE_DIR", None)
 
 
 def pytest_configure(config):

python/test/unit/runtime/test_cache.py

@@ -489,7 +489,7 @@ def cache_hook(*args, **kwargs):
     assert specialization_data is not None
 
     # clear the cache
-    shutil.rmtree(fresh_triton_cache, ignore_errors=True)
+    shutil.rmtree(fresh_triton_cache)
     kernel_add.device_caches[device][0].clear()
 
     # preload the kernel

python/test/unit/runtime/test_subproc.py

@@ -87,7 +87,7 @@ def test_compile_in_forked_subproc_with_forced_gc(fresh_triton_cache) -> None:
     compile_empty_kernel_with_gc()
 
     # stage 2.p
-    shutil.rmtree(fresh_triton_cache, ignore_errors=True)
+    shutil.rmtree(fresh_triton_cache)
     mp_ctx = multiprocessing.get_context(start_method)
     proc = mp_ctx.Process(target=compile_empty_kernel_with_gc)
 

third_party/intel/python/test/conftest.py

@@ -14,13 +14,9 @@ def device(request):
 
 @pytest.fixture
 def fresh_triton_cache():
-    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
+    with tempfile.TemporaryDirectory() as tmpdir:
+        try:
+            os.environ["TRITON_CACHE_DIR"] = tmpdir
+            yield tmpdir
+        finally:
+            os.environ.pop("TRITON_CACHE_DIR", None)