test(torchtitan): add qwen test

Author: Xiaoming-AMD

tests/trainer/test_torchtitan_trainer.py

@@ -145,15 +145,21 @@ def test_qwen3_0_6B(self):
             self.__class__.__name__,
             "qwen3_0.6B",
             "examples/torchtitan/configs/MI300X/qwen3_0.6B-pretrain.yaml",
-            extra_args=["--n_layers", "4", "--training.steps", "30"],
+            extra_args=["--model.n_layers", "4", "--training.steps", "3"],
         )
 
-    # def test_qwen3_1_7B(self):
-    #     run_script(self.__class__.__name__, "qwen3_1.7B",
-    #                "examples/torchtitan/configs/MI300X/qwen3_1.7B-pretrain.yaml",
-    #                extra_args=["--n_layers", "4", "--training.steps", "30"])
+    def test_qwen3_1_7B(self):
+        run_script(
+            self.__class__.__name__,
+            "qwen3_1.7B",
+            "examples/torchtitan/configs/MI300X/qwen3_1.7B-pretrain.yaml",
+            extra_args=["--model.n_layers", "4", "--training.steps", "3"],
+        )
 
-    # def test_qwen3_32B(self):
-    #     run_script(self.__class__.__name__, "qwen3_32B",
-    #               "examples/torchtitan/configs/MI300X/qwen3_32B-pretrain.yaml",
-    #               extra_args=["--n_layers", "4", "--training.steps", "30"])
+    def test_qwen3_32B(self):
+        run_script(
+            self.__class__.__name__,
+            "qwen3_32B",
+            "examples/torchtitan/configs/MI300X/qwen3_32B-pretrain.yaml",
+            extra_args=["--model.n_layers", "4", "--training.steps", "3"],
+        )