Skip to content

Commit b95b3e2

Browse files
committed
[recipe][perf] fix: format llama31 perf recipes
Signed-off-by: yaoyu-33 <yaoyu.094@gmail.com>
1 parent b18ee07 commit b95b3e2

3 files changed

Lines changed: 3 additions & 0 deletions

File tree

src/megatron/bridge/perf_recipes/llama/b200/llama31.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -137,6 +137,7 @@ def llama31_405b_pretrain_128gpu_b200_nvfp4_config() -> ConfigContainer:
137137

138138
llama31_405b_pretrain_256gpu_b200_fp8mx_config = llama31_405b_pretrain_256gpu_gb200_fp8mx_config
139139

140+
140141
def llama31_405b_pretrain_256gpu_b200_nvfp4_config() -> ConfigContainer:
141142
"""Llama3.1 405B pretrain: 256x B200, NVFP4."""
142143
cfg = llama31_405b_pretrain_256gpu_gb200_nvfp4_config()

src/megatron/bridge/perf_recipes/llama/b300/llama31.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -145,6 +145,7 @@ def llama31_405b_pretrain_256gpu_b300_bf16_config() -> ConfigContainer:
145145
cfg.model.gradient_accumulation_fusion = True
146146
return cfg
147147

148+
148149
llama31_405b_pretrain_256gpu_b300_fp8cs_config = llama31_405b_pretrain_256gpu_gb300_fp8cs_config
149150

150151
llama31_405b_pretrain_256gpu_b300_fp8mx_config = llama31_405b_pretrain_256gpu_gb300_fp8mx_config

src/megatron/bridge/perf_recipes/llama/vr200/llama31.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525

2626
llama31_405b_pretrain_256gpu_vr200_fp8mx_config = llama31_405b_pretrain_256gpu_gb200_fp8mx_config
2727

28+
2829
def llama31_405b_pretrain_256gpu_vr200_nvfp4_config() -> ConfigContainer:
2930
"""Llama3.1 405B pretrain: 256x VR200, NVFP4."""
3031
cfg = llama31_405b_pretrain_256gpu_gb200_nvfp4_config()

0 commit comments

Comments
 (0)