allow new leaked variable associated with torch.compile

speediedan · speediedan · commit 41ba76130bab · 2024-05-06T08:51:38.000-07:00
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -73,6 +73,7 @@ def restore_env_variables():
         "CRC32C_SW_MODE",  # leaked by tensorboardX
         "TRITON_CACHE_DIR",
         "OMP_NUM_THREADS",  # leaked by Lightning launchers
+        "_TORCHINDUCTOR_PYOBJECT_TENSOR_DATA_PTR",  # 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
`@@ -73,6 +73,7 @@ def restore_env_variables():`
`73`	`73`	`"CRC32C_SW_MODE", # leaked by tensorboardX`
`74`	`74`	`"TRITON_CACHE_DIR",`
`75`	`75`	`"OMP_NUM_THREADS", # leaked by Lightning launchers`
	`76`	`+ "_TORCHINDUCTOR_PYOBJECT_TENSOR_DATA_PTR", # leaked by torch.compile`
`76`	`77`	`}`
`77`	`78`	`leaked_vars.difference_update(allowlist)`
`78`	`79`	`assert not leaked_vars, f"test is leaking environment variable(s): {set(leaked_vars)}"`