From 14dc6bb87dce7f3d9e8de1d3959df7cbb5fa9e59 Mon Sep 17 00:00:00 2001 From: lijiachen19 Date: Fri, 28 Nov 2025 18:24:52 -0800 Subject: [PATCH] Fix(patch): fix patch for vllm-ascend https://github.com/volcengine/verl/issues/2564 --- .../vllm/patch/patch_funcs/v092/vllm_ascend_patch.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/ucm/integration/vllm/patch/patch_funcs/v092/vllm_ascend_patch.py b/ucm/integration/vllm/patch/patch_funcs/v092/vllm_ascend_patch.py index 6abccf1b..168b1188 100644 --- a/ucm/integration/vllm/patch/patch_funcs/v092/vllm_ascend_patch.py +++ b/ucm/integration/vllm/patch/patch_funcs/v092/vllm_ascend_patch.py @@ -40,6 +40,8 @@ def _enable_sparse() -> bool: def _apply_ascend_patch() -> None: """Apply patch for vLLM-Ascend.""" try: + from vllm_ascend.patch import platform, worker + if _enable_sparse(): _patch_attention_v1() _patch_mla_v1()