[CI][Bugfix] Add tp_shard_with_padding to padded MoE reload test mock (#57402)
Signed-off-by: kimi-no-na-wa <kimi-no-na-wa@users.noreply.github.com> Co-authored-by: kimi-no-na-wa <kimi-no-na-wa@users.noreply.github.com> Co-authored-by: Misha Goin <mgoin64@gmail.com>
vllm-agent committed
Sep 17, 2026 at 10:02 UTC
b2f2bd71ba6cff6d0678e2f83d45c1c08425144c
1 file changed
+1
tests/model_executor/model_loader/test_reload.py
+1
@@ -489,6 +489,7 @@ def test_padded_moe_reload_releases_each_layer(
489
is_act_and_mul=is_gated,
490
has_bias=has_bias,
491
tp_rank=tp_rank,
492
+ tp_shard_with_padding=False,
493
moe_parallel_config=SimpleNamespace(tp_size=2),
494
)
495
model = torch.nn.ModuleList()