From 0fab5946949712e86305c380ba0a88e060fe6c5b Mon Sep 17 00:00:00 2001 From: Bradley Dice Date: Fri, 4 Sep 2026 14:09:51 -0500 Subject: [PATCH] CI: migrate custom jobs to RTX PRO 6000 --- .github/workflows/build.yaml | 4 ++-- .github/workflows/pr.yaml | 10 +++++----- .github/workflows/test.yaml | 4 ++-- 3 files changed, 9 insertions(+), 9 deletions(-) diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml index bdd46b298f..217550f4ed 100644 --- a/.github/workflows/build.yaml +++ b/.github/workflows/build.yaml @@ -119,7 +119,7 @@ jobs: arch: "amd64" date: ${{ inputs.date }} container_image: "rapidsai/ci-conda:26.10-cuda${{ matrix.CUDA_VER }}-${{ matrix.LINUX_VER }}-py${{ matrix.PY_VER }}" - node_type: "gpu-l4-latest-1" + node_type: "gpu-rtxpro6000-latest-1" script: "ci/build_rust.sh" sha: ${{ inputs.sha }} go-build-matrix: @@ -152,7 +152,7 @@ jobs: arch: "amd64" date: ${{ inputs.date }} container_image: "rapidsai/ci-conda:26.10-cuda${{ matrix.CUDA_VER }}-${{ matrix.LINUX_VER }}-py${{ matrix.PY_VER }}" - node_type: "gpu-l4-latest-1" + node_type: "gpu-rtxpro6000-latest-1" script: "ci/build_go.sh" sha: ${{ inputs.sha }} java-build-matrix: diff --git a/.github/workflows/pr.yaml b/.github/workflows/pr.yaml index dc5a1e1852..e4fc63790e 100644 --- a/.github/workflows/pr.yaml +++ b/.github/workflows/pr.yaml @@ -589,7 +589,7 @@ jobs: matrix: ${{ fromJSON(needs.rocky8-clib-tests-matrix.outputs.matrix) }} with: build_type: pull-request - node_type: "gpu-l4-latest-1" + node_type: ${{ matrix.ARCH == 'amd64' && 'gpu-rtxpro6000-latest-1' || 'gpu-l4-latest-1' }} arch: "${{matrix.arch}}" date: ${{ inputs.date }}_c container_image: "rapidsai/ci-wheel:26.10-cuda${{ matrix.CUDA_VER }}-${{ matrix.LINUX_VER }}-py${{ matrix.PY_VER }}" @@ -622,7 +622,7 @@ jobs: matrix: ${{ fromJSON(needs.conda-java-build-and-tests-matrix.outputs.matrix) }} with: build_type: pull-request - node_type: "gpu-l4-latest-1" + node_type: "gpu-rtxpro6000-latest-1" arch: "amd64" container_image: "rapidsai/ci-conda:26.10-cuda${{ matrix.CUDA_VER }}-${{ matrix.LINUX_VER }}-py${{ matrix.PY_VER }}" script: "ci/test_java.sh" @@ -647,7 +647,7 @@ jobs: matrix: ${{ fromJSON(needs.conda-java-build-and-tests-matrix.outputs.matrix) }} with: build_type: pull-request - node_type: "gpu-l4-latest-1" + node_type: "gpu-rtxpro6000-latest-1" arch: "amd64" container_image: "rapidsai/ci-conda:26.10-cuda${{ matrix.CUDA_VER }}-${{ matrix.LINUX_VER }}-py${{ matrix.PY_VER }}" script: "ci/test_lucene.sh cuvs-java-cuda${{ matrix.CUDA_VER }}" @@ -680,7 +680,7 @@ jobs: matrix: ${{ fromJSON(needs.rust-build-matrix.outputs.matrix) }} with: build_type: pull-request - node_type: "gpu-l4-latest-1" + node_type: "gpu-rtxpro6000-latest-1" arch: "amd64" container_image: "rapidsai/ci-conda:26.10-cuda${{ matrix.CUDA_VER }}-${{ matrix.LINUX_VER }}-py${{ matrix.PY_VER }}" script: "ci/build_rust.sh" @@ -711,7 +711,7 @@ jobs: matrix: ${{ fromJSON(needs.go-build-matrix.outputs.matrix) }} with: build_type: pull-request - node_type: "gpu-l4-latest-1" + node_type: "gpu-rtxpro6000-latest-1" arch: "amd64" container_image: "rapidsai/ci-conda:26.10-cuda${{ matrix.CUDA_VER }}-${{ matrix.LINUX_VER }}-py${{ matrix.PY_VER }}" script: "ci/build_go.sh" diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml index c2ff71b3a2..3b8c38c4a6 100644 --- a/.github/workflows/test.yaml +++ b/.github/workflows/test.yaml @@ -98,7 +98,7 @@ jobs: branch: ${{ inputs.branch }} date: ${{ inputs.date }} sha: ${{ inputs.sha }} - node_type: "gpu-l4-latest-1" + node_type: "gpu-rtxpro6000-latest-1" arch: "amd64" container_image: "rapidsai/ci-conda:26.10-cuda${{ matrix.CUDA_VER }}-${{ matrix.LINUX_VER }}-py${{ matrix.PY_VER }}" script: "ci/test_java.sh" @@ -126,7 +126,7 @@ jobs: branch: ${{ inputs.branch }} date: ${{ inputs.date }} sha: ${{ inputs.sha }} - node_type: "gpu-l4-latest-1" + node_type: "gpu-rtxpro6000-latest-1" arch: "amd64" container_image: "rapidsai/ci-conda:26.10-cuda${{ matrix.CUDA_VER }}-${{ matrix.LINUX_VER }}-py${{ matrix.PY_VER }}" script: "ci/test_lucene.sh cuvs-java-cuda${{ matrix.CUDA_VER }}"