Skip to content

Commit 25a93b7

Browse files
jeejeeleeerictang000
authored andcommitted
[Bugfix] Fix torch.compile raise FileNotFoundError (vllm-project#15278)
Signed-off-by: Jee Jee Li <[email protected]>
1 parent 34bee0e commit 25a93b7

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

vllm/compilation/backends.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -399,6 +399,7 @@ def __call__(self, graph: fx.GraphModule, example_inputs) -> Callable:
399399
rank = vllm_config.parallel_config.rank
400400
dp_rank = vllm_config.parallel_config.data_parallel_rank
401401
local_cache_dir = os.path.join(cache_dir, f"rank_{rank}_{dp_rank}")
402+
os.makedirs(local_cache_dir, exist_ok=True)
402403
self.compilation_config.local_cache_dir = local_cache_dir
403404

404405
disable_cache = envs.VLLM_DISABLE_COMPILE_CACHE

0 commit comments

Comments
 (0)