We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f5710ef commit e3d8186Copy full SHA for e3d8186
vllm/compilation/decorators.py
@@ -403,8 +403,17 @@ def patched_inline_call(self_):
403
output = self.aot_compiled_fn(self, *args, **kwargs)
404
assert aot_compilation_path is not None
405
assert cache_dir is not None
406
- os.makedirs(cache_dir, exist_ok=True)
407
- self.aot_compiled_fn.save_compiled_function(aot_compilation_path)
+ try:
+ os.makedirs(cache_dir, exist_ok=True)
408
+ self.aot_compiled_fn.save_compiled_function(
409
+ aot_compilation_path
410
+ )
411
+ except Exception as e:
412
+ logger.warning(
413
+ "Cannot save aot compilation to path %s, error: %s",
414
+ aot_compilation_path,
415
+ str(e),
416
417
else:
418
output = self.compiled_callable(*args, **kwargs)
419
return output
0 commit comments