diff --git a/vllm_ascend/patch/worker/__init__.py b/vllm_ascend/patch/worker/__init__.py index e916aee811c..88fe563c44c 100644 --- a/vllm_ascend/patch/worker/__init__.py +++ b/vllm_ascend/patch/worker/__init__.py @@ -26,7 +26,6 @@ import vllm_ascend.patch.worker.patch_bert # noqa import vllm_ascend.patch.worker.patch_distributed # noqa import vllm_ascend.patch.worker.patch_multimodal_merge # noqa -import vllm_ascend.patch.worker.patch_rope # noqa import vllm_ascend.patch.worker.patch_qwen3_next # noqa import vllm_ascend.patch.worker.patch_qwen3_next_mtp # noqa import vllm_ascend.patch.worker.patch_rejection_sampler # noqa diff --git a/vllm_ascend/patch/worker/patch_rope.py b/vllm_ascend/patch/worker/patch_rope.py deleted file mode 100644 index ca4e7d1569c..00000000000 --- a/vllm_ascend/patch/worker/patch_rope.py +++ /dev/null @@ -1,31 +0,0 @@ -# -# Copyright (c) 2025 Huawei Technologies Co., Ltd. All Rights Reserved. -# This file is a part of the vllm-ascend project. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. -# - -import torch -import torch.nn as nn -from vllm.model_executor.layers.rotary_embedding.base import RotaryEmbeddingBase - - -class AscendRotaryEmbeddingBase(nn.Module): - def get_cos_sin(self, seqlen: int) -> tuple[torch.Tensor, torch.Tensor]: - cos_sin = self.cos_sin_cache[:seqlen] - cos, sin = cos_sin.chunk(2, dim=-1) - return cos, sin - - -# NOTE: These will be removed after vllm-ascend is aligned with vllm latest main. -RotaryEmbeddingBase.get_cos_sin = AscendRotaryEmbeddingBase.get_cos_sin