fix init_with_cudagraph_sizes

Signed-off-by: Icey <[email protected]>

Author: wxsIcey

vllm_ascend/platform.py

@@ -260,8 +260,11 @@ def check_and_update_config(cls, vllm_config: VllmConfig) -> None:
                 f"{vllm_config.parallel_config.tensor_parallel_size}")
             if len(sp_aclgraph_sizes) != len(original_sizes):
                 compilation_config.cudagraph_capture_sizes = sp_aclgraph_sizes
-                vllm_config.compilation_config.init_with_cudagraph_sizes(
-                    sp_aclgraph_sizes)
+                if vllm_version_is("0.11.0"):
+                    compilation_config.init_with_cudagraph_sizes(
+                        sp_aclgraph_sizes)
+                else:
+                    vllm_config.compilation_config.post_init_cudagraph_sizes()
 
         # TODO: Full graph is fully supported later, and the default value will be set to full graph.
         if compilation_config.cudagraph_mode == CUDAGraphMode.FULL_AND_PIECEWISE:

vllm_ascend/utils.py

@@ -402,7 +402,11 @@ def update_aclgraph_sizes(vllm_config: VllmConfig) -> None:
         indices[0], indices[-1] = 0, len(original_sizes) - 1
 
         sampled_sizes = [original_sizes[i] for i in indices]
-        compilation_config.init_with_cudagraph_sizes(sampled_sizes)
+        if vllm_version_is("0.11.0"):
+            compilation_config.init_with_cudagraph_sizes(sampled_sizes)
+        else:
+            compilation_config.cudagraph_capture_sizes = sampled_sizes
+            compilation_config.post_init_cudagraph_sizes()
 
         logger.info(
             "Adjusted ACL graph batch sizes for %s model (layers: %d): %d → %d sizes",
@@ -433,7 +437,11 @@ def update_aclgraph_sizes(vllm_config: VllmConfig) -> None:
         if original_sizes[0] < (num_speculative_tokens + 1) * max_num_seqs:
             enlarged_sizes = [(num_speculative_tokens + 1) * size
                               for size in original_sizes]
-            compilation_config.init_with_cudagraph_sizes(enlarged_sizes)
+            if vllm_version_is("0.11.0"):
+                compilation_config.init_with_cudagraph_sizes(enlarged_sizes)
+            else:
+                compilation_config.cudagraph_capture_sizes = enlarged_sizes
+                compilation_config.post_init_cudagraph_sizes()
             logger.info(
                 "Adjusted ACL graphs: %s → %s for speculative decoding",
                 original_sizes, enlarged_sizes)