xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-06-07 12:22:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix incorrect filenames in vllm_compile_cache.py (#15494)

Browse Source 
Signed-off-by: <zou3519@gmail.com>
Signed-off-by: youkaichao <youkaichao@gmail.com>
Co-authored-by: youkaichao <youkaichao@gmail.com>
This commit is contained in:
Richard Zou 2025-03-27 06:33:41 -04:00 committed by GitHub
parent 3f532cb6a6
commit 6278bc829e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
vllm/compilation/compiler_interface.py
View File

@ -229,7 +229,20 @@ class InductorAdaptor(CompilerInterface):
inductor_compiled_graph = output inductor_compiled_graph = output
if inductor_compiled_graph is not None: if inductor_compiled_graph is not None:
nonlocal file_path nonlocal file_path
file_path = inductor_compiled_graph.current_callable.__code__.co_filename # noqa compiled_fn = inductor_compiled_graph.current_callable
file_path = compiled_fn.__code__.co_filename # noqa
if not file_path.startswith(self.cache_dir):
# hooked in the align_inputs_from_check_idxs function
# in torch/_inductor/utils.py
for cell in compiled_fn.__closure__:
if not callable(cell.cell_contents):
continue
code = cell.cell_contents.__code__
if code.co_filename.startswith(self.cache_dir):
# this is the real file path
# compiled from Inductor
file_path = code.co_filename
break
hash_str = inductor_compiled_graph._fx_graph_cache_key hash_str = inductor_compiled_graph._fx_graph_cache_key
return output return output