File tree Expand file tree Collapse file tree 2 files changed +20
-1
lines changed
Expand file tree Collapse file tree 2 files changed +20
-1
lines changed Original file line number Diff line number Diff line change @@ -2,6 +2,13 @@ name: Slice-baseline-paddle
22
33on :
44 workflow_dispatch :
5+ inputs :
6+ PR_ID :
7+ required : false
8+ type : string
9+ COMMIT_ID :
10+ required : false
11+ type : string
512 schedule :
613 - cron : ' 0 20 * * 0'
714
4350 docker_build_image : ${{ needs.build-docker.outputs.docker_build_image }}
4451 slice-check : ' true'
4552 SLICE_TEST_MODE : insert_baseline
53+ MANUALLY_PR_ID : ${{ inputs.PR_ID }}
54+ MANUALLY_COMMIT_ID : ${{ inputs.COMMIT_ID }}
Original file line number Diff line number Diff line change 2020 type : string
2121 required : false
2222 default : ' paddle'
23+ MANUALLY_PR_ID :
24+ type : string
25+ required : false
26+ MANUALLY_COMMIT_ID :
27+ type : string
28+ required : false
2329
2430env :
2531 PR_ID : ${{ github.event.pull_request.number || '0' }}
@@ -105,7 +111,11 @@ jobs:
105111 if [[ "${{ inputs.SLICE_BENCHMARK_FRAMEWORKS }}" == "torch" ]];then
106112 python3.10 -m pip install torch==2.6.0 torchvision==0.21.0 torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cu118
107113 else
108- python3.10 -m pip install $wheel_link
114+ if [[ "${{ inputs.MANUALLY_PR_ID }}" == "" ]]; then
115+ python3.10 -m pip install $wheel_link
116+ else
117+ python3.10 -m pip install https://paddle-github-action.bj.bcebos.com/PR/build/${{ inputs.MANUALLY_PR_ID }}/${{ inputs.MANUALLY_COMMIT_ID }}/paddlepaddle_gpu-0.0.0-cp310-cp310-linux_x86_64.whl
118+ fi
109119 fi
110120 python3.10 -m pip install -r PaddleTest/framework/e2e/api_benchmark/requirement.txt
111121 cd PaddleTest/framework/slice_benchmark
You can’t perform that action at this time.
0 commit comments