Skip to content

Commit 55a29df

Browse files
committed
fix ut bug
Signed-off-by: whx-sjtu <2952154980@qq.com>
1 parent 37c030a commit 55a29df

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/ut/worker/test_model_runner_v1.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ def test_select_moe_comm_method(soc_version, enable_expert_parallel,
5757
mock_runner = MagicMock(spec=NPUModelRunner)
5858
mock_runner.parallel_config = MagicMock()
5959
mock_runner.parallel_config.enable_expert_parallel = enable_expert_parallel
60-
mock_runner.parallel_config.world_size = world_size
60+
mock_runner.parallel_config.world_size_across_dp = world_size
6161
mock_runner.mc2_tokens_capacity = mc2_tokens_capacity
6262

6363
# Patch the helper functions

0 commit comments

Comments
 (0)