Skip to content

Commit 9f174a5

Browse files
committed
Revert test_moe_deepseek_pipeline_subset to original mesh
1 parent 93e1b3e commit 9f174a5

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

tests/unit/train_compile_test.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -626,8 +626,8 @@ def test_moe_deepseek_pipeline_subset(self):
626626
"",
627627
get_test_config_path(),
628628
f"compiled_trainstep_file={compiled_trainstep_file}",
629-
"compile_topology=v5p-8",
630-
"compile_topology_num_slices=2",
629+
"compile_topology=v5p-64",
630+
"compile_topology_num_slices=8",
631631
"use_iota_embed=true",
632632
"model_name=deepseek3-test",
633633
"megablox=True",
@@ -636,8 +636,8 @@ def test_moe_deepseek_pipeline_subset(self):
636636
"per_device_batch_size=1",
637637
"max_target_length=1024",
638638
"pipeline_parallel_layers=56",
639-
"ici_expert_parallelism=8",
640-
"dcn_pipeline_parallelism=2",
639+
"ici_expert_parallelism=16",
640+
"dcn_pipeline_parallelism=8",
641641
)
642642
)
643643

0 commit comments

Comments
 (0)