Skip to content

Commit 1694d7c

Browse files
authored
Fix ci not set no_proxy and modify tests in pir mode (#9205)
1 parent 0289118 commit 1694d7c

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

scripts/distribute/ci_case_auto.sh

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -890,6 +890,7 @@ function llama_dy2st_auto_bs4_bf16_DP1-MP1-PP4-SD2-VPP3_split_bw() {
890890

891891
export CUDA_DEVICE_MAX_CONNECTIONS=1
892892
export PARALLEL_CROSS_ENTROPY=true
893+
export FLAGS_enable_pir_api=False # 暂时disable pir,后期修复后打开 @卢畅
893894

894895
task_name="llama_dy2st_auto_bs4_bf16_DP1-MP1-PP4-SD2-VPP3_split_bw"
895896
case_out_dir="output/$task_name"
@@ -2256,6 +2257,7 @@ function before_hook_for_gpt() {
22562257
env | grep FLAGS
22572258
export http_proxy=${proxy}
22582259
export https_proxy=${proxy}
2260+
export no_proxy=bcebos.com
22592261
if [[ $FLAGS_install_deps == 0 ]];then
22602262
echo -e "\033[31m ---- Install requirements for GPT auto cases \033[0m"
22612263
cp requirements.txt requirements_nlp.txt
@@ -2297,6 +2299,7 @@ function before_hook_for_llama() {
22972299
env | grep FLAGS
22982300
export http_proxy=${proxy}
22992301
export https_proxy=${proxy}
2302+
export no_proxy=bcebos.com
23002303
python -m pip install -r $root_path/requirements.txt
23012304
python -m pip install -r $root_path/requirements-dev.txt
23022305
if [[ ! $FLAGS_download_data =~ "llama" ]];then

scripts/distribute/ci_case_dy.sh

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ function gpt_case_list_dygraph(){
4242
gpt_generation_345M_single
4343
gpt_generation_345M_hybrid
4444
gpt_345M_mp8_qat
45-
gpt_export_345M_mp1
45+
# gpt_export_345M_mp1
4646
# gpt_export_345M_mp2
4747
# gpt_export_qat_345M
4848
# gpt_inference_345M_single
@@ -512,6 +512,7 @@ function before_hook_for_gpt() {
512512
env | grep FLAGS
513513
export http_proxy=${proxy}
514514
export https_proxy=${proxy}
515+
export no_proxy=bcebos.com
515516
if [[ $FLAGS_install_deps == 0 ]];then
516517
echo -e "\033[31m ---- Install requirements for GPT dygraph cases \033[0m"
517518
cp requirements.txt requirements_nlp.txt
@@ -619,6 +620,7 @@ function before_hook_for_llm_gpt() {
619620
env | grep FLAGS
620621
export http_proxy=${proxy}
621622
export https_proxy=${proxy}
623+
export no_proxy=bcebos.com
622624
python -m pip install -r $root_path/requirements.txt
623625
python -m pip install -r $root_path/requirements-dev.txt
624626
if [[ ! $FLAGS_download_data =~ "llm_gpt" ]];then

0 commit comments

Comments
 (0)