From 6711a81d6065fea68c472936a0f115ff12e06c13 Mon Sep 17 00:00:00 2001 From: Dennis Liu Date: Mon, 1 Dec 2025 23:33:37 -0800 Subject: [PATCH 1/3] Add new tests to github MoE test scope. --- tests/test_utils/recipes/moe.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_utils/recipes/moe.yaml b/tests/test_utils/recipes/moe.yaml index 285d16c99f3..cb29f36ddf1 100644 --- a/tests/test_utils/recipes/moe.yaml +++ b/tests/test_utils/recipes/moe.yaml @@ -170,12 +170,12 @@ products: - test_case: [gpt3_mcore_te_tp2_pp2_ep4_etp1_memory_speed] products: - environment: [dev] - scope: [mr] + scope: [mr, mr-github] platforms: [dgx_h100] - test_case: [gpt3_mcore_te_tp2_pp2_ep4_etp1_mtp_resume_torch_dist_fp8] products: - environment: [dev] - scope: [mr] + scope: [mr, mr-github] platforms: [dgx_h100] # hang: #513 - environment: [dev] scope: [mr-slim] From db36cf479a5f6b0dccbe5d3a7db60ba80b87e711 Mon Sep 17 00:00:00 2001 From: Dennis Liu Date: Tue, 2 Dec 2025 01:53:18 -0800 Subject: [PATCH 2/3] Update test cases. --- tests/test_utils/recipes/moe.yaml | 32 +++++++++++++++---------------- 1 file changed, 16 insertions(+), 16 deletions(-) diff --git a/tests/test_utils/recipes/moe.yaml b/tests/test_utils/recipes/moe.yaml index cb29f36ddf1..aea3ec97597 100644 --- a/tests/test_utils/recipes/moe.yaml +++ b/tests/test_utils/recipes/moe.yaml @@ -109,7 +109,7 @@ products: - test_case: [gpt3_mcore_te_tp2_zp_z3_resume_torch_dist_te_8experts2parallel_top2router] products: - environment: [dev] - scope: [mr] + scope: [mr, mr-github] platforms: [dgx_h100] - test_case: [gpt3_mcore_te_tp2_pp1_te_8experts2parallel_ddp_average_in_collective] products: @@ -121,30 +121,30 @@ products: - environment: [dev] scope: [mr] platforms: [dgx_h100] - # - test_case: [gpt3_moe_mcore_te_ep8_resume_torch_dist_dist_muon] - # products: - # - environment: [dev] - # scope: [mr, mr-github, mr-slim] - # platforms: [dgx_h100] - # - test_case: [gpt3_moe_mcore_te_ep8_resume_torch_dist_muon] - # products: - # - environment: [dev] - # scope: [mr, mr-github, mr-slim] - # platforms: [dgx_h100] + - test_case: [gpt3_moe_mcore_te_ep8_resume_torch_dist_dist_muon] + products: + - environment: [dev] + scope: [mr, mr-github] + platforms: [dgx_h100] + - test_case: [gpt3_moe_mcore_te_ep8_resume_torch_dist_muon] + products: + - environment: [dev] + scope: [mr, mr-github] + platforms: [dgx_h100] - test_case: [gpt3_moe_mcore_te_tp2_pp2_ep4_etp1_no_mtp_no_a2a_ovlp_fine_grained_offloading] products: - environment: [dev] - scope: [mr] + scope: [mr, mr-github] platforms: [dgx_h100] - test_case: [gpt3_moe_mcore_te_tp2_pp2_ep4_etp1_fine_grained_offloading] products: - environment: [dev] - scope: [mr] + scope: [mr, mr-github] platforms: [dgx_h100] - test_case: [gpt3_moe_mcore_te_tp4_ep2_etp2_pp2_scoped_cudagraph] products: - environment: [dev] - scope: [mr] + scope: [mr, mr-github] platforms: [dgx_h100] ####################################################################### # Super important mr, mr-github tests that run for both DEV and LTS per mr, mr-github # @@ -170,12 +170,12 @@ products: - test_case: [gpt3_mcore_te_tp2_pp2_ep4_etp1_memory_speed] products: - environment: [dev] - scope: [mr, mr-github] + scope: [mr] platforms: [dgx_h100] - test_case: [gpt3_mcore_te_tp2_pp2_ep4_etp1_mtp_resume_torch_dist_fp8] products: - environment: [dev] - scope: [mr, mr-github] + scope: [mr] platforms: [dgx_h100] # hang: #513 - environment: [dev] scope: [mr-slim] From 02c0aa9af738b24eb6600789fe22580216e64955 Mon Sep 17 00:00:00 2001 From: Dennis Liu Date: Tue, 2 Dec 2025 04:58:18 -0800 Subject: [PATCH 3/3] Change muon test to regular test type. --- .../model_config.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/functional_tests/test_cases/moe/gpt3_moe_mcore_te_ep8_resume_torch_dist_muon/model_config.yaml b/tests/functional_tests/test_cases/moe/gpt3_moe_mcore_te_ep8_resume_torch_dist_muon/model_config.yaml index 81b023bd86e..d3e3baa9f14 100644 --- a/tests/functional_tests/test_cases/moe/gpt3_moe_mcore_te_ep8_resume_torch_dist_muon/model_config.yaml +++ b/tests/functional_tests/test_cases/moe/gpt3_moe_mcore_te_ep8_resume_torch_dist_muon/model_config.yaml @@ -64,4 +64,4 @@ MODEL_ARGS: --muon-momentum: 0.9 --muon-extra-scale-factor: 0.2 --muon-scale-mode: spectral -TEST_TYPE: ckpt-resume +TEST_TYPE: regular