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
2
2
3
3
on :
4
4
workflow_dispatch :
5
+ inputs :
6
+ PR_ID :
7
+ required : false
8
+ type : string
9
+ COMMIT_ID :
10
+ required : false
11
+ type : string
5
12
schedule :
6
13
- cron : ' 0 20 * * 0'
7
14
43
50
docker_build_image : ${{ needs.build-docker.outputs.docker_build_image }}
44
51
slice-check : ' true'
45
52
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 20
20
type : string
21
21
required : false
22
22
default : ' paddle'
23
+ MANUALLY_PR_ID :
24
+ type : string
25
+ required : false
26
+ MANUALLY_COMMIT_ID :
27
+ type : string
28
+ required : false
23
29
24
30
env :
25
31
PR_ID : ${{ github.event.pull_request.number || '0' }}
@@ -105,7 +111,11 @@ jobs:
105
111
if [[ "${{ inputs.SLICE_BENCHMARK_FRAMEWORKS }}" == "torch" ]];then
106
112
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
107
113
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
109
119
fi
110
120
python3.10 -m pip install -r PaddleTest/framework/e2e/api_benchmark/requirement.txt
111
121
cd PaddleTest/framework/slice_benchmark
You can’t perform that action at this time.
0 commit comments