From 9f78b9ca8406fe0c120cefbf21deaa0b99b36d65 Mon Sep 17 00:00:00 2001
From: Jiangyun Zhu <riverclouds.zhu@qq.com>
Date: Mon, 29 Sep 2025 21:54:52 +0800
Subject: [PATCH] [torch.compile] serialize cudagraph_mode as its enum name
 instead of value (#25868)

Signed-off-by: zjy0516 <riverclouds.zhu@qq.com>
Signed-off-by: yewentao256 <zhyanwentao@126.com>
---
 vllm/config/compilation.py | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/vllm/config/compilation.py b/vllm/config/compilation.py
index 825de7d261918..df08fa20e0cfa 100644
--- a/vllm/config/compilation.py
+++ b/vllm/config/compilation.py
@@ -73,6 +73,9 @@ class CUDAGraphMode(enum.Enum):
             CUDAGraphMode.NONE, CUDAGraphMode.PIECEWISE, CUDAGraphMode.FULL
         ]
 
+    def __str__(self) -> str:
+        return self.name
+
 
 @config
 @dataclass
@@ -417,10 +420,11 @@ class CompilationConfig:
         if pass_config_exclude:
             exclude["pass_config"] = pass_config_exclude
 
-        return TypeAdapter(CompilationConfig).dump_json(
-            self,
-            exclude=exclude,  # type: ignore[arg-type]
-            exclude_unset=True).decode()
+        config = TypeAdapter(CompilationConfig).dump_python(self,
+                                                            exclude=exclude,
+                                                            exclude_unset=True)
+
+        return str(config)
 
     __str__ = __repr__