refactor(config): organize training configs by GPU architecture (MI300 / / MI355)

Xiaoming-AMD · Xiaoming-AMD · commit 0874a0d7ac76 · 2025-10-19T23:29:11.000-05:00
diff --git a/tests/configs/test_config.py b/tests/configs/test_config.py
@@ -37,7 +37,7 @@ def parse_config(self, cli_args: argparse.Namespace):
     def test_exp_configs(self):
         exps = [
             "examples/megatron/exp_pretrain.yaml",
-            "examples/torchtitan/configs/llama3.1_8B-BF16-pretrain.yaml",
+            "examples/torchtitan/configs/MI300X/llama3.1_8B-BF16-pretrain.yaml",
         ]
 
         for exp in exps:

Original file line number	Diff line number	Diff line change
`@@ -37,7 +37,7 @@ def parse_config(self, cli_args: argparse.Namespace):`
`37`	`37`	`def test_exp_configs(self):`
`38`	`38`	`exps = [`
`39`	`39`	`"examples/megatron/exp_pretrain.yaml",`
`40`		`- "examples/torchtitan/configs/llama3.1_8B-BF16-pretrain.yaml",`
	`40`	`+ "examples/torchtitan/configs/MI300X/llama3.1_8B-BF16-pretrain.yaml",`
`41`	`41`	`]`
`42`	`42`
`43`	`43`	`for exp in exps:`