Merge branch 'master' into sample_run

Doris26 · web-flow · commit a3204eb128f1 · 2025-04-21T10:48:21.000-07:00
diff --git a/dags/multipod/maxtext_v5e_configs_perf.py b/dags/multipod/maxtext_v5e_configs_perf.py
@@ -103,7 +103,7 @@
   for mode, image in DOCKER_IMAGES:
     for model in MaxTextV5eModelConfigs:
       base_run_model_cmds = [
-          f"python3 benchmarks/benchmark_runner.py on-device --base_output_directory={BASE_OUTPUT_DIRECTORY} --model_name={model.value} --libtpu_type=maxtext-docker --num_steps=15",
+          f"python3 benchmarks/benchmark_runner.py on-device --base_output_directory={BASE_OUTPUT_DIRECTORY} --model_name={model.value} --libtpu_type=maxtext-docker --num_steps=15 --use_pathways=True",
       ]
       maxtext_sweep_gke_test = (
           maxtext_sweep_gke_config.get_maxtext_sweep_gke_config(

Original file line number	Diff line number	Diff line change
`@@ -103,7 +103,7 @@`
`103`	`103`	`for mode, image in DOCKER_IMAGES:`
`104`	`104`	`for model in MaxTextV5eModelConfigs:`
`105`	`105`	`base_run_model_cmds = [`
`106`		`- f"python3 benchmarks/benchmark_runner.py on-device --base_output_directory={BASE_OUTPUT_DIRECTORY} --model_name={model.value} --libtpu_type=maxtext-docker --num_steps=15",`
	`106`	`+ f"python3 benchmarks/benchmark_runner.py on-device --base_output_directory={BASE_OUTPUT_DIRECTORY} --model_name={model.value} --libtpu_type=maxtext-docker --num_steps=15 --use_pathways=True",`
`107`	`107`	`]`
`108`	`108`	`maxtext_sweep_gke_test = (`
`109`	`109`	`maxtext_sweep_gke_config.get_maxtext_sweep_gke_config(`