Internet-Draft mcast4AI in Scale-Up July 2026
Wang, et al. Expires 20 January 2027 [Page]
Workgroup:
rtgwg
Internet-Draft:
draft-wang-mcast4ai-p2mp-scaleup-00
Published:
Intended Status:
Informational
Expires:
Authors:
H. Wang
Huawei
L. Zhang
Huawei
W. Yang
Huawei

AI P2MP Multicast in Scale-UP Network

Abstract

MoE (Mixture-of-Experts) has been adopted in scale-up GPU connections by the mainstream Large Language Models, such as Llama4, Mixtral, and DeepSeekV3. This document focuses on the AI multicast protocol evolution for MoE in scale-up scenarios, analyzing the challenges and proposing potential protocol evolutions.

Status of This Memo

This Internet-Draft is submitted in full conformance with the provisions of BCP 78 and BCP 79.

Internet-Drafts are working documents of the Internet Engineering Task Force (IETF). Note that other groups may also distribute working documents as Internet-Drafts. The list of current Internet-Drafts is at https://datatracker.ietf.org/drafts/current/.

Internet-Drafts are draft documents valid for a maximum of six months and may be updated, replaced, or obsoleted by other documents at any time. It is inappropriate to use Internet-Drafts as reference material or to cite them other than as "work in progress."

This Internet-Draft will expire on 20 January 2027.

Table of Contents

1. Introduction

The mainstream large language models, such as Llama4, Mixtral, and DeepSeekV3, have adopted MoE (Mixture-of-Experts) as the key technique. MoE has two procedures, dispatch and combine, to serve AI training and inference. The growing AlltoAll collective communication, performed during the dispatch procedure, has become the dominant in MoE, where tokens are required to be delivered to multiple selected expert nodes. This type of AI workload exhibits inherently point-to-multipoint (P2MP) communication patterns along with new requirements. Some valuable work has been done to address the requirements and gap analysis for supporting AI multicast communication. Referring to [I-D.zhang-rtgwg-multicast-requirements-gaps-aidc], it identifies the key requirements of multicast in AIDCs, including the particular ones for MoE. [I-D.mcbride-mcast4ai-p2mp-mechanism-evaluation] gives a gap analysis of the BIER protocol and proposes to introduce a new protocol. However, it has become a trend that MoE is primarily applied in scale-up scenarios where GPUs are connected in a more limited scale than scale-out scenarios. AI multicast in scale-up scenarios presents particular features, and the target optimization approaches should receive more attention. This document's scope is to make an analysis of the particular AI multicast features of MoE in scale-up scenarios, and give the recommended target multicast protocol evolution approaches.

2. Requirements of multicast communication supporting MoE

Several requirements of multicast communication supporting MoE have been addressed in [I-D.zhang-rtgwg-multicast-requirements-gaps-aidc], including:

Networks in AIDCs are required to support closed-loop control for MoE, including efficient P2MP multicast forwarding and P2MP forwarding for feedback acknowledgments (ACKs). Moreover, it is required to support MP2P packet aggregation in large AIDCs with numerous receivers to reduce the excessive reverse traffic that can lead to network congestion.

MoE-based AI training and inference use token dispatch, where gating networks select expert nodes per token at microsecond timescales, dynamically determining real-time multicast receiver sets with no fixed groups. This ultra-fast selection leaves no time for traditional multicast to establish, update, or tear down trees, leading to delays, packet loss, or AI task failure [I-D.zhang-rtgwg-llmmoe-multicast]. Therefore, multicast in AIDC should meet high dynamics requirements, including fast change of multicast members and low overhead for dynamic change in both the control plane and data plane.

Multicast in AIDCs frequently involves multicast groups where only a small fraction of the total nodes in the cluster are multicast members, a characteristic closely tied to the sparse activation mechanism of modern AI models such as MoE. For example, DeepSeekV3 uses 256 experts and activates 9 experts at a time. The multicast should be efficient when the group size is small relative to the network size, and meet the sparseness requirements, including efficient sparse member identification to avoid unnecessary scanning or signaling of non-member nodes, and low overhead for sparse state maintenance to reduce state maintenance burden and ensure no additional latency during state updates.

Besides these requirements, there is one more: memory address mapping. For LLMs, RDMA write operations require specifying the memory addresses of remote GPUs for data writes. In MoE dispatch, the randomly selected experts reside on different destination nodes, each with a distinct memory address. Since the dispatch process inherently involves sending data to a multicast group, the address mapping requirement refers to an efficient method for conveying the address information of these multicast group nodes to the source node.

3. Differences between MoE in scale-out and scale-up scenarios

MoE in scale-out and scale-up scenarios exhibits different features.

With the growth of LLMs in parameter size, the scale-out network is predicted to reach a cluster of 128K nodes. Although the scale‑up network keeps enlarging, its scale is far smaller than that of the scale‑out network, no more than 1K nodes.

In scale‑out scenarios, message semantics (Send/Receive) are primarily used in RDMA communication, whereas memory semantics (RDMA Read/Write) is the primary mode in scale‑up scenarios.

4. Challenges for AI multicast communication of MoE in scale-out and scale-up scenarios

Since there are critical differences between MoE in scale-out and scale-up scenarios given in Section 3, the challenges of AI multicast protocol evolution also differ.

In scale-out scenarios, ACK problems should be comprehensively considered to solve, because ACK is the key process within the message semantics mode of RDMA communication. Referring to [I-D.mcbride-mcast4ai-p2mp-mechanism-evaluation], solving ACK aggregation problems requires significant new machinery that may increase the complexity. However, in scale-up scenarios, these problems no longer exist under the memory semantics mode.

Since high dynamics exist in both scale-out and scale-up scenarios, the challenges to introduce BIER-like protocols are the same.

Since scale‑out scenarios may involve thousands of endpoints, far larger than the size of a single sub‑domain, hierarchical domains become a critical implementation requirement in AI multicast protocol evolution. In contrast, scale‑up scenarios are typically limited to no more than 1K nodes, where a single sub‑domain suffices. As a result, the complexity of AI multicast protocol evolution in scale‑up scenarios is much lower than that in scale‑out scenarios, and can therefore yield greater performance gains.

Memory address mapping problems exist in both scale-out and scale-up scenarios, so that the approach is the same.

As the scale-up scenarios are increasingly adopted by the mainstream LLMs, meeting the AI multicast requirements in this scenario becomes more urgent and valuable.

5. Target multicast protocol evolutions in scale-up network

BIER's stateless forwarding model aligns well with AI P2MP groups whose membership is known at the source, as is the case in MoE; the subset of active expert endpoints varies per training step, even though the full set of experts is fixed for the duration of the job. However, evolutions of BIER to support MoE are still necessary.

Beyond these target evolutions of AI multicast protocols, the memory address mapping problems remain unresolved. Addressing the question of how to efficiently convey multiple destination addresses persists as an inescapable requirement that cannot be circumvented.

6. Open Issues

While the dispatch phase can be independently optimized—for instance, through compact XPU-ID lists or lightweight acknowledgment mechanisms—such isolated improvements offer only marginal benefits. A balanced, co-optimized approach that synchronously addresses both dispatch and combine is essential to achieve meaningful end-to-end performance improvements.

7. Security Considerations

This document does not introduce any new security considerations.

8. IANA Considerations

This document has no IANA actions.

9. Informative References

10. Informative References

[I-D.zhang-rtgwg-multicast-requirements-gaps-aidc]
Zhang, J., Cheng, W., and K. Liu, "Requirements and Gap Analysis of Multicast in AI Data Centers", Work in Progress, Internet-Draft, draft-zhang-rtgwg-multicast-requirements-gaps-aidc-02, , <https://datatracker.ietf.org/doc/html/draft-zhang-rtgwg-multicast-requirements-gaps-aidc-02>.
[I-D.mcbride-mcast4ai-p2mp-mechanism-evaluation]
McBride, M., Liu, Y., and L. Zhang, "AI P2MP Mechanism Evaluation", Work in Progress, Internet-Draft, draft-mcbride-mcast4ai-p2mp-mechanism-evaluation-00, , <https://datatracker.ietf.org/doc/html/draft-mcbride-mcast4ai-p2mp-mechanism-evaluation-00>.
[RFC8279]
Wijnands, IJ., Ed., Rosen, E., Ed., Dolganow, A., Przygienda, T., and S. Aldrin, "Multicast Using Bit Index Explicit Replication (BIER)", RFC 8279, DOI 10.17487/RFC8279, , <https://www.rfc-editor.org/rfc/rfc8279>.
[I-D.zhang-rtgwg-llmmoe-multicast]
Zhang, Z., Duan, W., Xu, X., and Y. Liu, "Multicast use case in LLM MoE", Work in Progress, Internet-Draft, draft-zhang-rtgwg-llmmoe-multicast-02, , <https://datatracker.ietf.org/doc/html/draft-zhang-rtgwg-llmmoe-multicast-02>.

Contributors

Authors' Addresses

Haibo Wang
Huawei
Li Zhang
Huawei
Wenbin Yang
Huawei