Modify runs-on labels

nod-ai · Dec 5, 2024 · 50f8279 · 50f8279
1 parent d12d701
commit 50f8279
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 8 deletions.
diff --git a/.github/workflows/ci_gpu.yml b/.github/workflows/ci_gpu.yml
@@ -18,10 +18,11 @@ jobs:
   ci-gpu:
     runs-on:
       - self-hosted
+      - ci-gpu
       - ${{ matrix.gpu-type }}]
     strategy:
       matrix:
-        gpu-type: [ci-gpu-cuda, ci-gpu-rocm]
+        gpu-type: [cuda, rocm]
     steps:
       - name: Clean up previous run
         run: |
@@ -47,11 +48,11 @@ jobs:
         with:
           submodules: recursive
       - name: Install dependencies for CUDA
-        if: matrix.gpu-type == 'ci-gpu-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 == 'ci-gpu-rocm'
+        if: matrix.gpu-type == 'rocm'
         run: |
           python ts_scripts/install_dependencies.py --environment=dev --rocm=rocm62
       - name: Torchserve Sanity

diff --git a/.github/workflows/kserve_gpu_tests.yml b/.github/workflows/kserve_gpu_tests.yml
@@ -10,10 +10,11 @@ jobs:
   kserve-gpu-tests:
     runs-on:
       - self-hosted
-      - ${{ matrix.gpu-type }}]
+      - regression-test-gpu
+      - ${{ matrix.gpu-type }}
     strategy:
       matrix:
-        gpu-type: [regression-test-gpu-cuda, regression-test-gpu-rocm]
+        gpu-type: [cuda, rocm]
     steps:
       - name: Clean up previous run
         run: |

diff --git a/.github/workflows/regression_tests_gpu.yml b/.github/workflows/regression_tests_gpu.yml
@@ -19,10 +19,11 @@ jobs:
     # creates workflows on self hosted runner
     runs-on:
       - self-hosted
+      - regression-test-gpu
       - ${{ matrix.gpu-type }}
     strategy:
       matrix:
-        gpu-type: [regression-test-gpu-cuda, regression-test-gpu-rocm]
+        gpu-type: [cuda, rocm]
     steps:
       - name: Clean up previous run
         run: |
@@ -50,11 +51,11 @@ jobs:
         with:
           submodules: recursive
       - name: Install dependencies for CUDA
-        if: matrix.gpu-type == 'regression-test-gpu-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 == 'regression-test-gpu-rocm'
+        if: matrix.gpu-type == 'rocm'
         run: |
           python ts_scripts/install_dependencies.py --environment=dev --rocm=rocm62
       - name: Torchserve Regression Tests