Skip to content

Commit a77f0c9

Browse files
committed
fix ut
Signed-off-by: MengqingCao <cmq0113@163.com>
1 parent 3e1d2bb commit a77f0c9

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/ut/ops/test_linear.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,10 @@ def setUp(self):
3333
return_value=self.mock_group),
3434
patch("vllm_ascend.ops.linear_op.get_tp_group",
3535
return_value=self.mock_group),
36+
patch(
37+
"vllm.distributed.parallel_state.get_tp_group",
38+
return_value=self.mock_group,
39+
),
3640
patch("vllm_ascend.utils.mlp_tp_enable", return_value=True),
3741
patch("vllm_ascend.utils.oproj_tp_enable", return_value=True)
3842
]

0 commit comments

Comments
 (0)