From 2ba68d9bf704434c8eb9364fefc4459819347077 Mon Sep 17 00:00:00 2001 From: Yongye Zhu Date: Wed, 10 Jun 2026 12:43:12 -0400 Subject: [PATCH] [Test] Fix one-sided MNNVL alltoall test workspace under-reservation (#44946) Signed-off-by: Yongye Zhu --- tests/distributed/test_mnnvl_alltoall.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tests/distributed/test_mnnvl_alltoall.py b/tests/distributed/test_mnnvl_alltoall.py index 53fea072555..95c905fc080 100644 --- a/tests/distributed/test_mnnvl_alltoall.py +++ b/tests/distributed/test_mnnvl_alltoall.py @@ -747,6 +747,11 @@ def _one_sided_data_worker(rank, world_size): top_k=experts_per_token, num_experts=num_experts, hidden_size=hidden_size, + # Account for the fp8 block-scale payload (a1q_scale: hidden//16 bytes + # per token) that is dispatched alongside the nvfp4 hidden states. + # Without this the dispatch region is under-reserved and the combine + # payload overflows the per-rank workspace. + dispatch_scale_bytes_per_token=hidden_size // 16, ) assert manager.initialized assert manager.moe_alltoall is not None