disable use_cinn for intermediate_api_meta-llama-Llama-2-7b (#10778)

XieYunshen · web-flow · commit b8f4b0cba6ce · 2025-06-27T20:27:10.000+08:00
diff --git a/tests/test_tipc/static/auto_parallel/llama2/N4C32/intermediate_api_meta-llama-Llama-2-7b_pretrain_dy2st_bs32_bf16_Sharding32_Stage2.sh b/tests/test_tipc/static/auto_parallel/llama2/N4C32/intermediate_api_meta-llama-Llama-2-7b_pretrain_dy2st_bs32_bf16_Sharding32_Stage2.sh
@@ -19,6 +19,7 @@ param+="global_batch_size=32 "
 param+="nnodes=4 "
 param+="model_type=llama2_7b "
 param+='intermediate_api=intermediate_api_ '
+export FLAGS_use_cinn=0
 
 cd ./tests
 bash ./test_tipc/static/auto_parallel/llama2/benchmark_common/prepare.sh
diff --git a/tests/test_tipc/static/auto_parallel/llama2/benchmark_common/run_benchmark.sh b/tests/test_tipc/static/auto_parallel/llama2/benchmark_common/run_benchmark.sh
@@ -261,7 +261,7 @@ export CUDA_DEVICE_MAX_CONNECTIONS=1
 export PARALLEL_CROSS_ENTROPY=true
 
 # cinn相关
-export FLAGS_use_cinn=1
+export FLAGS_use_cinn=${FLAGS_use_cinn:-1}
 export FLAGS_dist_prim_all=1
 export FLAGS_prim_forward_blacklist="pd_op.stack;pd_op.squeeze;pd_op.swiglu;pd_op.squared_l2_norm"
 export FLAGS_prim_backward_blacklist="swiglu_grad"
