Skip to content

Commit 846e596

Browse files
committed
fix bug in DenseOptimRowParallelOp when tp=1
Signed-off-by: realliujiaxu <realliujiaxu@163.com>
1 parent 53ecd89 commit 846e596

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

vllm_ascend/ops/linear_op.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -385,7 +385,9 @@ def apply(
385385
bias_ = None if (self.tp_rank > 0 or self.skip_bias_add) else self.bias
386386

387387
if self.tp_size == 1 or not self.reduce_results:
388-
output = self.quant_method.apply(self, input_parallel, bias=bias_)
388+
output = self.quant_method.apply(self.layer,
389+
input_parallel,
390+
bias=bias_)
389391
else:
390392
output_parallel = self.quant_method.apply(self.layer,
391393
input_parallel,

0 commit comments

Comments
 (0)