On Windows use Triton cache dir per pytest worker (#3041)

Fixes #2777.
intel · Dec 20, 2024 · 00858c3 · 00858c3
1 parent 97ba60e
commit 00858c3
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/python/test/unit/conftest.py b/python/test/unit/conftest.py
@@ -24,3 +24,10 @@ def fresh_triton_cache():
     except OSError:
         # Ignore errors, such as PermissionError, on Windows
         pass
+
+
+def pytest_configure(config):
+    worker_id = os.getenv("PYTEST_XDIST_WORKER")
+    # On Windows, use a dedicated Triton cache per pytest worker to avoid PermissionError.
+    if os.name == "nt" and worker_id:
+        os.environ["TRITON_CACHE_DIR"] = tempfile.mkdtemp(prefix="triton-")