Merge commit '145dc25de45aea13ddf582ed8cda7c8c1be6a8ff'

Author: whitneywhtsang

python/test/unit/conftest.py

@@ -70,11 +70,10 @@ def device(request):
 @pytest.fixture
 def fresh_triton_cache():
     with tempfile.TemporaryDirectory() as tmpdir:
-        try:
-            os.environ["TRITON_CACHE_DIR"] = tmpdir
+        from triton import knobs
+        with knobs.cache.scope():
+            knobs.cache.dir = tmpdir
             yield tmpdir
-        finally:
-            os.environ.pop("TRITON_CACHE_DIR", None)
 
 
 def _fresh_knobs_impl(monkeypatch, skipped_attr: Optional[Set[str]] = None):