serve/.github/workflows/ci_gpu.yml at 301266873fc38eac1e2f2299d872ac9fcc47341f · pytorch/serve · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
name: CI GPU

on:
  workflow_dispatch:
  push:
    branches:
      - master
  pull_request:
    branches:
      - master
  merge_group:

concurrency:
  group: ci-gpu-${{ github.workflow }}-${{ github.ref == 'refs/heads/master' && github.run_number || github.ref }}
  cancel-in-progress: true

jobs:
  ci-gpu:
    runs-on: [self-hosted, ci-gpu]
    runs-on:
      - self-hosted
      - ci-gpu
      - ${{ matrix.gpu-type }}
    strategy:
      matrix:
        gpu-type: [cuda, rocm]
    steps:
      - name: Clean up previous run
        run: |
          echo "Cleaning up previous run"
          ls -la ./
          # deletes all files in the current directory
          sudo rm -rf ./* || true
          # deletes all hidden files in a directory
          sudo rm -rf ./.??* || true
          ls -la ./
      - name: Setup Python 3.9
        uses: actions/setup-python@v5
        with:
          python-version: 3.9
          architecture: x64
      - name: Setup Java 17
        uses: actions/setup-java@v3
        with:
          distribution: 'zulu'
          java-version: '17'
      - name: Checkout TorchServe
        uses: actions/checkout@v3
        with:
          submodules: recursive
      - name: Install dependencies for CUDA
        if: matrix.gpu-type == 'cuda'
        run: |
          python ts_scripts/install_dependencies.py --environment=dev --cuda=cu121
      - name: Install dependencies for ROCm
        if: matrix.gpu-type == 'rocm'
        run: |
          python ts_scripts/install_dependencies.py --environment=dev --rocm=rocm6.2
      - name: Torchserve Sanity
        uses: nick-fields/retry@v3
        with:
          timeout_minutes: 60
          retry_on: error
          max_attempts: 3
          command: |
            python torchserve_sanity.py

      # Any coverage.xml will be picked up by this step
      # Just make sure each coverage.xml is in a different folder
      - name: Upload codecov
        run : |
          curl -Os https://uploader.codecov.io/latest/linux/codecov
          chmod +x codecov
          ./codecov