Merge branch 'master' into dev/cpu/init_data_optimisation

Author: trivialfis

.github/runs-on.yml

@@ -22,20 +22,30 @@ runners:
     cpu: 16
     family: ["c7i-flex", "c7i", "c7a", "c5", "c5a"]
     image: linux-amd64
+    spot: "false"
   linux-amd64-gpu:
     family: ["g4dn.xlarge"]
     image: linux-amd64
+    spot: "false"
   linux-amd64-mgpu:
     family: ["g4dn.12xlarge"]
     image: linux-amd64
+    spot: "false"
   linux-arm64-cpu:
     cpu: 16
     family: ["c6g", "c7g"]
     image: linux-arm64
+    spot: "false"
+  linux-arm64-gpu:
+    family: ["g5g.xlarge"]
+    image: linux-arm64
+    spot: "false"
   windows-gpu:
     family: ["g4dn.2xlarge"]
     image: windows-amd64
+    spot: "false"
   windows-cpu:
     cpu: 32
     family: ["c7i-flex", "c7i", "c7a", "c5", "c5a"]
     image: windows-amd64
+    spot: "false"

.github/workflows/cccl_nightly.yml

@@ -0,0 +1,82 @@
+name: Test XGBoost with latest CCCL and RMM
+
+on:
+  workflow_dispatch:  # Can be manually triggered
+  schedule:
+    - cron: "0 7 * * *"  # Run once daily
+
+permissions:
+  contents: read
+
+defaults:
+  run:
+    shell: bash -l {0}
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
+  cancel-in-progress: true
+
+env:
+  GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+
+jobs:
+  get-latest-cccl-version:
+    name: Query the latest version of CCCL
+    runs-on: ubuntu-latest
+    outputs:
+      cccl_version: ${{ steps.query_version_step.outputs.cccl_version }}
+    steps:
+      - uses: actions/checkout@v4
+      - id: query_version_step
+        name: Query version
+        run: |
+          source ops/pipeline/query-latest-cccl.sh
+          echo "cccl_version=${CCCL_VERSION}" >> "$GITHUB_OUTPUT"
+
+  test-latest-cccl:
+    name: Test building XGBoost with latest CCCL (RC included)
+    runs-on:
+      - runs-on=${{ github.run_id }}
+      - runner=linux-amd64-cpu
+      - tag=nightly-cccl
+    needs: get-latest-cccl-version
+    steps:
+      - uses: actions/checkout@v4
+        with:
+          submodules: "true"
+      - name: Log into Docker registry (AWS ECR)
+        run: bash ops/pipeline/login-docker-registry.sh
+      - name: Build XGBoost with latest CCCL
+        run: |
+          bash ops/pipeline/nightly-test-cccl.sh ${{ needs.get-latest-cccl-version.outputs.cccl_version }}
+
+  get-latest-rmm-version:
+    name: Query the latest version of RMM
+    runs-on: ubuntu-latest
+    outputs:
+      rmm_version: ${{ steps.query_version_step.outputs.rmm_version }}
+    steps:
+      - uses: actions/checkout@v4
+      - id: query_version_step
+        name: Query version
+        run: |
+          source ops/pipeline/query-latest-rmm.sh
+          echo "rmm_version=${RMM_VERSION}" >> "$GITHUB_OUTPUT"
+
+  test-latest-rmm:
+    name: Test building XGBoost with latest nightly version of RMM
+    # This job uses the stable CCCL used by RMM and rest of RAPIDS
+    runs-on:
+      - runs-on=${{ github.run_id }}
+      - runner=linux-amd64-cpu
+      - tag=nightly-rmm
+    needs: get-latest-rmm-version
+    steps:
+      - uses: actions/checkout@v4
+        with:
+          submodules: "true"
+      - name: Log into Docker registry (AWS ECR)
+        run: bash ops/pipeline/login-docker-registry.sh
+      - name: Build XGBoost with latest RMM
+        run: |
+          bash ops/pipeline/nightly-test-rmm.sh ${{ needs.get-latest-rmm-version.outputs.rmm_version }}

.github/workflows/doc.yml

@@ -15,12 +15,12 @@ jobs:
     timeout-minutes: 20
     name: A job to run test in FreeBSD
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v6.0.1
         with:
           submodules: 'true'
       - name: Test in FreeBSD
         id: test
-        uses: vmactions/freebsd-vm@v1
+        uses: vmactions/freebsd-vm@v1.3.6
         with:
           usesh: true
           prepare: |