TORCHINDUCTOR_CACHE_DIR

Borda · Borda · commit bc3cbd97ce6c · 2025-01-07T20:38:52.000+09:00
diff --git a/tests/tests_pytorch/conftest.py b/tests/tests_pytorch/conftest.py
@@ -93,6 +93,7 @@ def restore_env_variables():
         "TF_CPP_MIN_LOG_LEVEL",
         "TF_GRPC_DEFAULT_OPTIONS",
         "XLA_FLAGS",
+        "TORCHINDUCTOR_CACHE_DIR", # leaked by torch.compile
     }
     leaked_vars.difference_update(allowlist)
     assert not leaked_vars, f"test is leaking environment variable(s): {set(leaked_vars)}"

Original file line number	Diff line number	Diff line change
`@@ -93,6 +93,7 @@ def restore_env_variables():`
`93`	`93`	`"TF_CPP_MIN_LOG_LEVEL",`
`94`	`94`	`"TF_GRPC_DEFAULT_OPTIONS",`
`95`	`95`	`"XLA_FLAGS",`
	`96`	`+ "TORCHINDUCTOR_CACHE_DIR", # leaked by torch.compile`
`96`	`97`	`}`
`97`	`98`	`leaked_vars.difference_update(allowlist)`
`98`	`99`	`assert not leaked_vars, f"test is leaking environment variable(s): {set(leaked_vars)}"`