From e7296b08da66b4c79eb43d0932a3d8628178d036 Mon Sep 17 00:00:00 2001 From: Angela Yi Date: Fri, 5 Dec 2025 08:54:26 -0800 Subject: [PATCH] [bugfix] Pass globals to aot_compiled function (#29428) Signed-off-by: angelayi --- vllm/compilation/decorators.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/vllm/compilation/decorators.py b/vllm/compilation/decorators.py index eed7795cdb349..6bb66ce3eec0c 100644 --- a/vllm/compilation/decorators.py +++ b/vllm/compilation/decorators.py @@ -409,7 +409,9 @@ def _support_torch_compile( open(aot_compilation_path, "rb") as f, ): start_monitoring_torch_compile(self.vllm_config) - loaded_fn = torch.compiler.load_compiled_function(f) + loaded_fn = torch.compiler.load_compiled_function( + f, f_globals=self.forward.__globals__ + ) _verify_source_unchanged(loaded_fn.source_info(), self.vllm_config) loaded_fn.disable_guard_check() self.aot_compiled_fn = loaded_fn