torch (#74227)

Author: swgu98

.github/workflows/Slice-baseline.yml

@@ -3,7 +3,7 @@ name: Slice-baseline
 on:
   workflow_dispatch:
   schedule:
-    - cron: '30 2 * * *'
+    - cron: '0 20 * * *'
 
 permissions: read-all
 

.github/workflows/Slice-torch.yml

@@ -0,0 +1,38 @@
+name: Slice-baseline
+
+on:
+  workflow_dispatch:
+  schedule:
+    - cron: '0 0 1 * *'
+
+permissions: read-all
+
+defaults:
+  run:
+    shell: bash
+
+jobs:
+  clone:
+    name: Slice-base clone
+    uses: ./.github/workflows/_Clone-linux.yml
+    with:
+      clone_dir: Paddle-build
+      is_pr: 'false'
+
+  build-docker:
+    name: Slice build docker
+    needs: clone
+    uses: ./.github/workflows/docker.yml
+    with:
+      clone_dir: Paddle-build
+      task: build
+
+  slice-test:
+    name: Slice test
+    uses: ./.github/workflows/_Slice.yml
+    needs: [clone, build-docker]
+    with:
+      docker_build_image: ${{ needs.build-docker.outputs.docker_build_image }}
+      slice-check: 'true'
+      SLICE_TEST_MODE: insert_baseline
+      SLICE_BENCHMARK_FRAMEWORKS: torch

.github/workflows/_Slice.yml

@@ -16,6 +16,10 @@ on:
         type: string
         required: false
         default: 'test_ci'
+      SLICE_BENCHMARK_FRAMEWORKS:
+        type: string
+        required: false
+        default: 'paddle'
 
 env:
   PR_ID: ${{ github.event.pull_request.number || '0' }}
@@ -44,6 +48,7 @@ jobs:
           wheel_link: https://paddle-github-action.bj.bcebos.com/PR/build/${{ env.PR_ID }}/${{ env.COMMIT_ID }}/paddlepaddle_gpu-0.0.0-cp310-cp310-linux_x86_64.whl
           RUN_ID: ${{ github.run_id }}
           SLICE_TEST_MODE: ${{ inputs.SLICE_TEST_MODE }}
+          SLICE_BENCHMARK_FRAMEWORKS: ${{ inputs.SLICE_BENCHMARK_FRAMEWORKS }}
         run: |
           container_name="api_benchmark_ci_${RUN_ID}"
           echo "container_name=${container_name}" >> ${{ github.env }}
@@ -64,6 +69,7 @@ jobs:
             -e no_proxy \
             -e CI_name \
             -e SLICE_TEST_MODE \
+            -e SLICE_BENCHMARK_FRAMEWORKS \
             -w /paddle --network host ${docker_image}
 
       - name: Download PaddleTest
@@ -88,11 +94,14 @@ jobs:
           mkdir -p ${{ github.workspace }}/../../../.cache/pip
           source ${{ github.workspace }}/../../../proxy
           python3.10 -m pip config set global.cache-dir ${{ github.workspace }}/../../../.cache/pip
-          python3.10 -m pip install $wheel_link
+          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
+          fi
           python3.10 -m pip install -r PaddleTest/framework/e2e/api_benchmark/requirement.txt
           cd PaddleTest/framework/slice_benchmark
           cp ${{ github.workspace }}/../../../apibm_config.yml .
-          nvidia-smi
           bash run.sh
           '