2222 description : " Enable nightly build mode (e.g. add date suffix to version)"
2323 required : false
2424 type : string
25- default : " ON "
25+ default : " OFF "
2626 FD_VERSION :
2727 description : " FastDeploy Package Version"
2828 required : false
2929 type : string
3030 default : " "
31+ PADDLEVERSION :
32+ description : " Paddle Version Build Use"
33+ required : false
34+ type : string
35+ default : " "
36+ PADDLE_WHL_URL :
37+ description : " Paddle Wheel Package URL"
38+ required : false
39+ type : string
40+ default : " "
3141 UPLOAD :
3242 description : " Upload Package"
3343 required : false
8595 compile_arch : ${{ inputs.COMPILE_ARCH }}
8696 fd_version : ${{ inputs.FD_VERSION }}
8797 CACHE_DIR : ${{ inputs.CACHE_DIR }}
98+ BRANCH_REF : ${{ github.ref_name }}
99+ PADDLEVERSION : ${{ inputs.PADDLEVERSION }}
100+ PADDLE_WHL_URL : ${{ inputs.PADDLE_WHL_URL }}
101+ WITH_NIGHTLY_BUILD : ${{ inputs.WITH_NIGHTLY_BUILD }}
88102 run : |
89103 set -x
90104 runner_name="${{ runner.name }}"
@@ -109,6 +123,9 @@ jobs:
109123 -e "COMPILE_ARCH=${compile_arch}" \
110124 -e "FD_VERSION=${fd_version}" \
111125 -e "WITH_NIGHTLY_BUILD=${WITH_NIGHTLY_BUILD}" \
126+ -e "PADDLEVERSION=${PADDLEVERSION}" \
127+ -e "PADDLE_WHL_URL=${PADDLE_WHL_URL}" \
128+ -e "BRANCH_REF=${BRANCH_REF}" \
112129 --gpus "\"device=${gpu_id}\"" ${docker_image} /bin/bash -c '
113130 if [[ -n "${FD_VERSION}" ]]; then
114131 export FASTDEPLOY_VERSION=${FD_VERSION}
@@ -124,7 +141,15 @@ jobs:
124141 echo "Date Only: $DATE_ONLY"
125142 export FASTDEPLOY_VERSION="${FASTDEPLOY_VERSION}.dev${DATE_ONLY}"
126143 fi
127- python -m pip install --pre paddlepaddle-gpu -i https://www.paddlepaddle.org.cn/packages/nightly/cu126/
144+ # 针对不同分支和tag使用不同的PaddlePaddle安装包
145+ if [[ "${PADDLE_WHL_URL}" != "" ]];then
146+ python -m pip install ${PADDLE_WHL_URL}
147+ elif [[ "${PADDLEVERSION}" != "" ]];then
148+ python -m pip install paddlepaddle-gpu==${PADDLEVERSION} -i https://www.paddlepaddle.org.cn/packages/stable/cu126/
149+ else
150+ python -m pip install --pre paddlepaddle-gpu -i https://www.paddlepaddle.org.cn/packages/nightly/cu126/
151+ fi
152+
128153 pip config set global.index-url https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple
129154
130155 python -m pip install --upgrade pip
0 commit comments