xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-24 01:55:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

[torch.compile] serialize cudagraph_mode as its enum name instead of value (#25868)

Browse Source 
Signed-off-by: zjy0516 <riverclouds.zhu@qq.com>
Signed-off-by: yewentao256 <zhyanwentao@126.com>
This commit is contained in:
Jiangyun Zhu 2025-09-29 21:54:52 +08:00 committed by yewentao256
parent 4e2774f5c3
commit 9f78b9ca84
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
vllm/config/compilation.py
View File

@ -73,6 +73,9 @@ class CUDAGraphMode(enum.Enum):
CUDAGraphMode.NONE, CUDAGraphMode.PIECEWISE, CUDAGraphMode.FULL CUDAGraphMode.NONE, CUDAGraphMode.PIECEWISE, CUDAGraphMode.FULL
] ]
def __str__(self) -> str:
return self.name
@config @config
@dataclass @dataclass
@ -417,10 +420,11 @@ class CompilationConfig:
if pass_config_exclude: if pass_config_exclude:
exclude["pass_config"] = pass_config_exclude exclude["pass_config"] = pass_config_exclude
return TypeAdapter(CompilationConfig).dump_json( config = TypeAdapter(CompilationConfig).dump_python(self,
self, exclude=exclude,
exclude=exclude, # type: ignore[arg-type] exclude_unset=True)
exclude_unset=True).decode()
return str(config)
__str__ = __repr__ __str__ = __repr__