perf-expert-parallel-overlap

Validates and tunes expert-parallel communication overlap for Megatron-Bridge MoE training on GPU clusters.

852|445|Updated May 21, 2025
One-click install
npx skills add https://github.com/NVIDIA-NeMo/Megatron-Bridge --skill perf-expert-parallel-overlap
Or copy as Structured Prompt for Agent▼
Please help me install this Agent Skill.
Skill: perf-expert-parallel-overlap
Source: https://github.com/NVIDIA-NeMo/Megatron-Bridge/tree/main/skills/perf-expert-parallel-overlap
Command: npx skills add https://github.com/NVIDIA-NeMo/Megatron-Bridge --skill perf-expert-parallel-overlap

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill includes scripts (resource) and references (resource) components.

What problem does it solve?

This Skill enhances the efficiency of MoE models by overlapping expert-parallel communication with computation, reducing step time.

Core Features & Use Cases

  • Communication Overlap: Validates and enables expert-parallel communication overlap during training.
  • Performance Tuning: Assists in tracing throughput regressions and tuning EP overlap configurations.
  • Use Case: Optimizing a large-scale MoE training run on GPU clusters to decrease latency and improve overall throughput with minimal code changes.

Quick Start

Use the performance skill to enable expert-parallel communication overlap during training by configuring overlap_moe_expert_parallel_comm and running the setup script.

Frequently Asked Questions about perf-expert-parallel-overlap

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I overlap expert-parallel communication with computation in MoE training?▼

To overlap expert-parallel communication in MoE training, you can enable the overlap_moe_expert_parallel_comm configuration and run the setup script to validate and apply the optimization for reduced training step times.

Why is my MoE training throughput regressing on GPU clusters?▼

MoE training throughput regressions often stem from unoptimized expert-parallel communication. This Skill traces performance bottlenecks and tunes EP overlap configurations like alltoall and flex dispatchers to improve GPU utilization.

Can I use alltoall and flex dispatchers for MoE communication overlap in Megatron-Bridge?▼

Yes, Megatron-Bridge supports key configurations like alltoall and flex dispatchers. This Skill validates these expert-parallel communication overlap settings to ensure safe deployment and improved throughput.

Do I need a specific PyTorch version to enable MoE communication overlap?▼

You need a compatible PyTorch version and sufficient memory constraints to safely deploy MoE communication overlap. This Skill checks your environment's PyTorch version and memory limits before applying the optimization.

What is expert-parallel communication overlap and when do I need it for deep learning?▼

Expert-parallel communication overlap is a performance tuning technique that overlaps MoE network operations with computation. You need it during large-scale MoE training on GPU clusters to reduce latency and decrease step times.

How to tune MoE communication overlap configurations with minimal code changes?▼

You can tune MoE communication overlap configurations by using this performance Skill to enable the overlap_moe_expert_parallel_comm setting. It validates your setup and applies optimizations with minimal code changes required.