From ce66812cb8ecb69e369f5c3a7139c7531aa4d4d4 Mon Sep 17 00:00:00 2001 From: Lihao Ran Date: Wed, 14 May 2025 18:05:42 +0000 Subject: [PATCH] Update config path --- benchmarks/mlperf/scripts/config_utils.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/benchmarks/mlperf/scripts/config_utils.sh b/benchmarks/mlperf/scripts/config_utils.sh index 866cd585..b194d004 100755 --- a/benchmarks/mlperf/scripts/config_utils.sh +++ b/benchmarks/mlperf/scripts/config_utils.sh @@ -9,7 +9,7 @@ export compute_axis_order=${compute_axis_order:=0,2,1,3} export prefill_cache_axis_order=${prefill_cache_axis_order:=0,2,1,3} export ar_cache_axis_order=${ar_cache_axis_order:=0,2,1,3} -export config_file_path=MaxText/configs/v5e/inference/llama2_70b_v5e-16.yml +export config_file_path=MaxText/inference/configs/multi_host/interleaved/llama2_70b_v5e-16.yml export ici_fsdp_parallelism=1 export ici_autoregressive_parallelism=${ici_autoregressive_parallelism:=2} export ici_tensor_parallelism=${ici_tensor_parallelism:=8}