Skip to content

Commit 66a7d6e

Browse files
committed
fix typo
Signed-off-by: realliujiaxu <[email protected]>
1 parent 61bae75 commit 66a7d6e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm_ascend/ops/linear.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -442,7 +442,7 @@ def get_custom_tp_group_row(
442442
if custom_tp_group is not None:
443443
return custom_tp_group, custom_tp_group.tp_rank, custom_tp_group.tp_size
444444

445-
return None, get_tp_group().tp_rank, get_tp_group().tp_size
445+
return None, get_tp_group().rank_in_group, get_tp_group().world_size
446446

447447

448448
class AscendColumnParallelLinear(ColumnParallelLinear):

0 commit comments

Comments
 (0)