fix(recipe): enable VR in perf configs#4724
Conversation
- gpt_oss 120B / qwen3 235B & 30B: apply full-iteration CUDA-graph configs on VR200 (and mxfp8 fp8_output_proj for gpt_oss 120B) - perf_plugins/executors: add vr200 to the GB200/GB300 platform lists (NVL72 topology, cuDNN LayerNorm, sm_100+ overlap, NUMA divisor) Signed-off-by: Md Fahim Faysal Khan <mdfahimfaysa@nvidia.com>
|
/claude review |
|
Light review — LGTM Clean, consistent VR200 enablement. VR200 is treated the same as GB200/GB300 (NVL72 topology, 4 GPUs/node so numa_divisor=2, SM100 dispatcher, cuDNN-LN branch), matching the existing base-config aliases and argument_parser.py (vr200: 4). The new/updated vr200 config functions mirror their gb200/gb300 siblings, including the fp8_mx guard on set_full_iter_cg_configs and the mxfp8 fp8_output_proj toggle for gpt_oss_120b. Observations (non-blocking):
Suggested test cases (perf configs touched: qwen3_235b_a22b vr200, qwen3_30b_a3b vr200, gpt_oss_120b vr200):
Each should instantiate the corresponding pretrain_config_vr200(precision=...) function and assert it builds without error (and that the fp8_mx path sets the full-iter CG / fp8_output_proj fields). |
|
/ok to test 6a5ba94 |
What does this PR do ?
Fixes few VR perf configs
Changelog
GitHub Actions CI
See the CI section in the Contributing doc for how to trigger the CI. A Nvidia developer will need to approve and trigger the CI for external contributors.
Before your PR is "Ready for review"
Pre checks: