Skip to content

Commit a7d6f1d

Browse files
authored
update core number (GoogleCloudPlatform#837)
1 parent 067c33b commit a7d6f1d

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

dags/multipod/maxtext_configs_aot.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -109,10 +109,10 @@
109109
)
110110

111111
# GPU AoT tests
112-
cmd = f"bash MaxText/configs/a3/llama_2_7b/16vm.sh EXECUTABLE=train_compile M_COMPILE_TOPOLOGY=a3 M_COMPILE_TOPOLOGY_NUM_SLICES=16"
112+
cmd = "bash MaxText/configs/a3/llama_2_7b/8vm.sh EXECUTABLE=train_compile M_COMPILE_TOPOLOGY=a3 M_COMPILE_TOPOLOGY_NUM_SLICES=8"
113113
stable_a3_gpu = gke_config.get_maxtext_end_to_end_gpu_gke_test_config(
114114
time_out_in_min=300,
115-
test_name=f"maxtext-aot-a3-stable",
115+
test_name="maxtext-aot-a3-stable",
116116
run_model_cmds=(cmd,),
117117
num_slices=1,
118118
cluster=XpkClusters.GPU_A3PLUS_CLUSTER,

0 commit comments

Comments
 (0)