mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-05-25 16:31:18 +08:00
[compile] Disable dynamo guards check for AOT compilation. (#27288)
Signed-off-by: zhxchen17 <zhxchen17@fb.com> Co-authored-by: Cyrus Leung <tlleungac@connect.ust.hk>
This commit is contained in:
parent
05181cc57f
commit
a00d6254e9
@ -302,6 +302,7 @@ def _support_torch_compile(
|
|||||||
start_monitoring_torch_compile(self.vllm_config)
|
start_monitoring_torch_compile(self.vllm_config)
|
||||||
loaded_fn = torch.compiler.load_compiled_function(f)
|
loaded_fn = torch.compiler.load_compiled_function(f)
|
||||||
_verify_source_unchanged(loaded_fn.source_info(), self.vllm_config)
|
_verify_source_unchanged(loaded_fn.source_info(), self.vllm_config)
|
||||||
|
loaded_fn.disable_guard_check()
|
||||||
self.aot_compiled_fn = loaded_fn
|
self.aot_compiled_fn = loaded_fn
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
if os.path.exists(aot_compilation_path):
|
if os.path.exists(aot_compilation_path):
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user