From 650add70e812bb47fb44bc35ad3bfe11e185fe3d Mon Sep 17 00:00:00 2001 From: gushiwei Date: Wed, 6 Aug 2025 09:38:25 +0000 Subject: [PATCH] add slice manual with pr --- .github/workflows/Slice-baseline.yml | 9 +++++++++ .github/workflows/_Slice.yml | 12 +++++++++++- 2 files changed, 20 insertions(+), 1 deletion(-) diff --git a/.github/workflows/Slice-baseline.yml b/.github/workflows/Slice-baseline.yml index ca544625a45eb9..4ab346a7a2a4dc 100644 --- a/.github/workflows/Slice-baseline.yml +++ b/.github/workflows/Slice-baseline.yml @@ -2,6 +2,13 @@ name: Slice-baseline-paddle on: workflow_dispatch: + inputs: + PR_ID: + required: false + type: string + COMMIT_ID: + required: false + type: string schedule: - cron: '0 20 * * 0' @@ -43,3 +50,5 @@ jobs: docker_build_image: ${{ needs.build-docker.outputs.docker_build_image }} slice-check: 'true' SLICE_TEST_MODE: insert_baseline + MANUALLY_PR_ID: ${{ inputs.PR_ID }} + MANUALLY_COMMIT_ID: ${{ inputs.COMMIT_ID }} diff --git a/.github/workflows/_Slice.yml b/.github/workflows/_Slice.yml index 8f74843fb313e9..9f5b671134ed09 100644 --- a/.github/workflows/_Slice.yml +++ b/.github/workflows/_Slice.yml @@ -20,6 +20,12 @@ on: type: string required: false default: 'paddle' + MANUALLY_PR_ID: + type: string + required: false + MANUALLY_COMMIT_ID: + type: string + required: false env: PR_ID: ${{ github.event.pull_request.number || '0' }} @@ -105,7 +111,11 @@ jobs: if [[ "${{ inputs.SLICE_BENCHMARK_FRAMEWORKS }}" == "torch" ]];then 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 else - python3.10 -m pip install $wheel_link + if [[ "${{ inputs.MANUALLY_PR_ID }}" == "" ]]; then + python3.10 -m pip install $wheel_link + else + 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 + fi fi python3.10 -m pip install -r PaddleTest/framework/e2e/api_benchmark/requirement.txt cd PaddleTest/framework/slice_benchmark