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 37c030a commit 55a29dfCopy full SHA for 55a29df
tests/ut/worker/test_model_runner_v1.py
@@ -57,7 +57,7 @@ def test_select_moe_comm_method(soc_version, enable_expert_parallel,
57
mock_runner = MagicMock(spec=NPUModelRunner)
58
mock_runner.parallel_config = MagicMock()
59
mock_runner.parallel_config.enable_expert_parallel = enable_expert_parallel
60
- mock_runner.parallel_config.world_size = world_size
+ mock_runner.parallel_config.world_size_across_dp = world_size
61
mock_runner.mc2_tokens_capacity = mc2_tokens_capacity
62
63
# Patch the helper functions
0 commit comments