gtest shouldn't depend on RMM dev

dmlc · Nov 26, 2024 · d450733 · d450733
1 parent 2e2ccec
commit d450733
Showing 1 changed file with 28 additions and 14 deletions.
diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
@@ -132,20 +132,12 @@ jobs:
           done
 
   build-cuda-with-rmm:
-    name: Build CUDA with RMM (${{ matrix.config.name }})
+    name: Build CUDA with RMM
     needs: build-containers
     runs-on:
       - runs-on=${{ github.run_id }}
       - runner=linux-amd64-cpu
-      - tag=main-build-cuda-with-rmm-${{ matrix.config.name }}
-    strategy:
-      fail-fast: false
-      matrix:
-        config:
-        - name: stable
-          container_id: xgb-ci.gpu_build_rockylinux8
-        - name: dev
-          container_id: xgb-ci.gpu_dev_ver
+      - tag=main-build-cuda-with-rmm
     steps:
       # Restart Docker daemon so that it recognizes the ephemeral disks
       - run: sudo systemctl restart docker
@@ -155,23 +147,45 @@ jobs:
       - name: Fetch container from cache
         run: bash ops/docker_build.sh
         env:
-          CONTAINER_ID: ${{ matrix.config.container_id }}
+          CONTAINER_ID: xgb-ci.gpu_build_rockylinux8
       - name: Fetch container from cache
         run: bash ops/docker_build.sh
         env:
           CONTAINER_ID: xgb-ci.manylinux_2_28_x86_64
       - run: |
-          bash ops/pipeline/build-cuda-with-rmm.sh ${{ matrix.config.container_id }}
+          bash ops/pipeline/build-cuda-with-rmm.sh xgb-ci.gpu_build_rockylinux8
       - name: Stash files
         run: |
           bash ops/pipeline/stash-artifacts.sh  \
             stash build-cuda-with-rmm build/testxgboost
-        if: matrix.config.name == 'stable'
       - name: Upload Python wheel
         run: |
           bash ops/pipeline/publish-artifact.sh python-package/dist/*.whl \
             s3://xgboost-nightly-builds/experimental_build_with_rmm/
-        if: matrix.config.name == 'stable'
+
+  build-cuda-with-rmm-dev:
+    name: Build CUDA with RMM (dev)
+    needs: build-containers
+    runs-on:
+      - runs-on=${{ github.run_id }}
+      - runner=linux-amd64-cpu
+      - tag=main-build-cuda-with-rmm-dev
+    steps:
+      # Restart Docker daemon so that it recognizes the ephemeral disks
+      - run: sudo systemctl restart docker
+      - uses: actions/checkout@v4
+        with:
+          submodules: "true"
+      - name: Fetch container from cache
+        run: bash ops/docker_build.sh
+        env:
+          CONTAINER_ID: xgb-ci.gpu_dev_ver
+      - name: Fetch container from cache
+        run: bash ops/docker_build.sh
+        env:
+          CONTAINER_ID: xgb-ci.manylinux_2_28_x86_64
+      - run: |
+          bash ops/pipeline/build-cuda-with-rmm.sh xgb-ci.gpu_dev_ver
 
   build-manylinux2014:
     name: Build manylinux2014_${{ matrix.arch }} wheel