From 203b4e67773d74ff196cb42d33a716dab939f83c Mon Sep 17 00:00:00 2001 From: zzhxxx Date: Mon, 1 Dec 2025 19:57:01 +0800 Subject: [PATCH] [Bug_fix] fix torchair o_proj forward parameter (#4166) ### What this PR does / why we need it? In `torchair_mla.py`, the `self.oproj` function includes an additional parameter `is_force_scatter`, while the `AscendRowParallelLinear` function in `linear.py` does not add this parameter. - vLLM version: v0.11.2 - vLLM main: https://github.com/vllm-project/vllm/commit/v0.11.2 Signed-off-by: zzhx1 --- vllm_ascend/ops/linear.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm_ascend/ops/linear.py b/vllm_ascend/ops/linear.py index 3b6f7100..688520a3 100644 --- a/vllm_ascend/ops/linear.py +++ b/vllm_ascend/ops/linear.py @@ -299,7 +299,7 @@ class AscendRowParallelLinear(RowParallelLinear): def forward( self, input_, - is_prefill: bool = True, + **kwargs, ) -> Union[torch.Tensor, tuple[torch.Tensor, Optional[Parameter]]]: if self.custom_op is not None: return self.custom_op.apply(input_)