[Misc] Drop patch_rope.py (#6291)
### What this PR does / why we need it?
Part of #5304.
We have align with vLLM's latest change for `RotaryEmbeddingBase`. Don't
need this patch anymore.
### Does this PR introduce _any_ user-facing change?
### How was this patch tested?
- vLLM version: v0.14.1
- vLLM main:
dc917cceb8
Signed-off-by: gcanlin <canlinguosdu@gmail.com>
This commit is contained in:
@@ -26,7 +26,6 @@ import vllm_ascend.patch.worker.patch_unquantized_gemm # noqa
|
|||||||
import vllm_ascend.patch.worker.patch_bert # noqa
|
import vllm_ascend.patch.worker.patch_bert # noqa
|
||||||
import vllm_ascend.patch.worker.patch_distributed # noqa
|
import vllm_ascend.patch.worker.patch_distributed # noqa
|
||||||
import vllm_ascend.patch.worker.patch_multimodal_merge # 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 # noqa
|
||||||
import vllm_ascend.patch.worker.patch_qwen3_next_mtp # noqa
|
import vllm_ascend.patch.worker.patch_qwen3_next_mtp # noqa
|
||||||
import vllm_ascend.patch.worker.patch_rejection_sampler # noqa
|
import vllm_ascend.patch.worker.patch_rejection_sampler # noqa
|
||||||
|
|||||||
@@ -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
|
|
||||||
Reference in New Issue
Block a user