Skip to content
This repository was archived by the owner on May 20, 2026. It is now read-only.

Commit 2612caa

Browse files
committed
Update perf reproduce instructions
Signed-off-by: Parth Mannan <pmannan@nvidia.com>
1 parent 9eaace1 commit 2612caa

1 file changed

Lines changed: 10 additions & 5 deletions

File tree

examples/megatron/recipes/wan/README_perf_test.md

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ This guide provides concise steps to set up the environment and run WAN pretrain
55
## Container Launch
66

77
```bash
8-
CONT="nvcr.io/nvidia/nemo:25.09.00"
8+
CONT="nvcr.io/nvidia/nemo:25.11"
99
MOUNT="/lustre/fsw/:/lustre/fsw/"
1010

1111
srun -t 02:00:00 \
@@ -28,18 +28,18 @@ cd /opt/
2828

2929
# DFM (pinned)
3030
git clone --no-checkout https://github.com/NVIDIA-NeMo/DFM.git
31-
git -C DFM checkout 174bb7b34de002ebbbcae1ba8e2b12363c7dee01
31+
git -C DFM checkout 9eaace14995a724c982fe53726a909be2edc93cb
3232
export DFM_PATH=/opt/DFM
3333

3434
# Megatron-Bridge (pinned)
3535
rm -rf /opt/Megatron-Bridge
36-
git clone --no-checkout https://github.com/huvunvidia/Megatron-Bridge.git
37-
git -C Megatron-Bridge checkout 713ab548e4bfee307eb94a7bb3f57c17dbb31b50
36+
git clone --no-checkout https://github.com/NVIDIA-NeMo/Megatron-Bridge.git
37+
git -C Megatron-Bridge checkout 953aabf75c0500180dc14a6a76cf9e7e7c4baec7
3838

3939
# Megatron-LM (pinned)
4040
rm -rf /opt/Megatron-LM
4141
git clone --no-checkout https://github.com/NVIDIA/Megatron-LM.git
42-
git -C Megatron-LM checkout ce8185cbbe04f38beb74360e878450f2e8525885
42+
git -C Megatron-LM checkout 2d398b42fd4237fffb553109563d73ac099751c3
4343

4444
# Python path
4545
export PYTHONPATH="${DFM_PATH}/.:/opt/Megatron-Bridge/.:/opt/Megatron-LM"
@@ -143,6 +143,11 @@ NVTE_FUSED_ATTN=1 torchrun --nproc_per_node=8 examples/megatron/recipes/wan/pret
143143
- Using `--mock` argument.
144144
- Adjust `video_size` (F_latents, H_latents, W_latents) and `number_packed_samples` of `WanMockDataModuleConfig` in `wan.py`. Total `seq_len = F * H * W * number_packed_samples`.
145145

146+
### Reproducing performance recipes
147+
148+
- Please use the appropriate system config recipe in `examples/megatron/recipes/wan/conf/<h100/gb200/gb300>_perf_pretrain_mock.yaml`
149+
- Usage example `examples/megatron/recipes/wan/pretrain_wan.py --mock --training-mode pretrain --config-file examples/megatron/recipes/wan/conf/gb300_perf_pretrain_mock.yaml`
150+
146151
## Inference
147152

148153
```bash

0 commit comments

Comments
 (0)