We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3e1d2bb commit a77f0c9Copy full SHA for a77f0c9
tests/ut/ops/test_linear.py
@@ -33,6 +33,10 @@ def setUp(self):
33
return_value=self.mock_group),
34
patch("vllm_ascend.ops.linear_op.get_tp_group",
35
36
+ patch(
37
+ "vllm.distributed.parallel_state.get_tp_group",
38
+ return_value=self.mock_group,
39
+ ),
40
patch("vllm_ascend.utils.mlp_tp_enable", return_value=True),
41
patch("vllm_ascend.utils.oproj_tp_enable", return_value=True)
42
]
0 commit comments