We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 067c33b commit a7d6f1dCopy full SHA for a7d6f1d
1 file changed
dags/multipod/maxtext_configs_aot.py
@@ -109,10 +109,10 @@
109
)
110
111
# 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"
+ cmd = "bash MaxText/configs/a3/llama_2_7b/8vm.sh EXECUTABLE=train_compile M_COMPILE_TOPOLOGY=a3 M_COMPILE_TOPOLOGY_NUM_SLICES=8"
113
stable_a3_gpu = gke_config.get_maxtext_end_to_end_gpu_gke_test_config(
114
time_out_in_min=300,
115
- test_name=f"maxtext-aot-a3-stable",
+ test_name="maxtext-aot-a3-stable",
116
run_model_cmds=(cmd,),
117
num_slices=1,
118
cluster=XpkClusters.GPU_A3PLUS_CLUSTER,
0 commit comments