onboard-gb200-1node-tests

Generate single-node GB200 test cases from 2-node MR recipes.

17.4k|4.3k|Updated Mar 21, 2019
One-click install
npx skills add https://github.com/NVIDIA/Megatron-LM --skill onboard-gb200-1node-tests-nvidia
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: onboard-gb200-1node-tests
Source: https://github.com/NVIDIA/Megatron-LM/tree/main/skills/onboard-gb200-1node-tests
Command: npx skills add https://github.com/NVIDIA/Megatron-LM --skill onboard-gb200-1node-tests-nvidia

SYSTEM DOCUMENTATION & REQUIREMENTS

What problem does it solve?

It solves the problem of expanding CI coverage for NVIDIA GB200 by turning existing 2-node MR-scoped functional tests into correct 1-node (mr-github) variants.

Core Features & Use Cases

  • Generates single-node test cases by scanning gpt.yaml and moe.yaml for candidate mr/mr-slim entries and excluding those already present as *-1node.yaml.
  • Adapts parallelism settings for 8→4 GPU runs by reading each candidate model_config.yaml and applying safe PP/EP reductions (while keeping global batch settings unchanged).
  • Creates or updates the required recipe files and product scopes by producing gpt-1node.yaml entries and/or appending to moe-1node.yaml, then wiring scope: [mr-github, mr-github-slim] where appropriate.

Quick Start

Run the onboard flow to scan existing GB200 2-node MR tests and produce their corresponding 1-node (_1node) model_config.yaml directories plus the matching gpt-1node.yaml/moe-1node.yaml recipe entries.

Frequently Asked Questions about onboard-gb200-1node-tests

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

FAQPage Schema
How do I generate single-node test variants from existing 2-node Megatron recipes?

Creating 1-node GitHub MR functional tests involves deriving single-node variants from existing 2-node mr-scoped recipes by scanning gpt.yaml and moe.yaml, then adjusting GPU parallelism for world_size reduction from 8 to 4 without changing the global batch size.

How does parallelism tuning work when reducing GB200 GPU runs from 8 to 4?

Parallelism tuning for 8 to 4 GPU runs reads each candidate model_config.yaml and applies safe PP/EP reductions to adapt the TP/PP/EP/ETP/DP relationships, ensuring the global batch settings remain completely unchanged during the world_size reduction.

Do I need to manually update recipe product scopes when onboarding 1-node GB200 tests?

You do not need to manually update product scopes when onboarding 1-node GB200 tests, because the process automatically produces gpt-1node.yaml entries or appends to moe-1node.yaml and wires scope: [mr-github, mr-github-slim] where appropriate.

What is the process for expanding CI coverage for GB200 GPT and MoE models?

Expanding CI coverage for GB200 GPT and MoE models requires scanning recipe product specs in gpt.yaml and moe.yaml for candidate mr and mr-slim entries, excluding those already present as *-1node.yaml, and generating the missing *_1node test cases.

Why are some existing 2-node MR tests excluded during 1-node test generation?

Existing 2-node MR tests are excluded during 1-node test generation if matching *-1node.yaml files are already present, preventing duplicate recipe entries and ensuring only missing *_1node test cases are generated for the CI expansion.