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.