[compile] Nest inductor cache under AOT compile dir (#39718)

Signed-off-by: Alessandro Sangiorgi <asangior@redhat.com>
This commit is contained in:
Alessandro Sangiorgi
2026-04-14 17:17:54 +00:00
committed by GitHub
parent 23f3760217
commit 2faad08362
+10
View File
@@ -507,6 +507,16 @@ def _support_torch_compile(
hash_key,
)
# Hash-level dir; shared across ranks on the same node.
self.compilation_config.local_cache_dir = cache_dir
inductor_cache = os.path.join(cache_dir, "inductor_cache")
os.makedirs(inductor_cache, exist_ok=True)
# Process-wide: post-load execution, CUDA-graph capture, and later
# autotune/recompile all need to write under {hash}/inductor_cache/.
# Unconditional because torch's cache_dir() may have pre-filled the
# /tmp default during import, making setdefault a no-op.
os.environ["TORCHINDUCTOR_CACHE_DIR"] = inductor_cache
rank = self.vllm_config.parallel_config.rank
dp_rank = self.vllm_config.parallel_config.data_parallel_index
cache_dir = os.path.join(cache_dir, f"rank_{rank}_{dp_rank}")