Skip to content

Remove deprecated ext_intel_global_device_space (#82) #31

Remove deprecated ext_intel_global_device_space (#82)

Remove deprecated ext_intel_global_device_space (#82) #31

Workflow file for this run

name: XGBoost CI (CUDA 13)
on: [push, pull_request]
permissions:
contents: read # to fetch code (actions/checkout)
concurrency:
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
cancel-in-progress: true
env:
BRANCH_NAME: >-
${{ github.event.pull_request.number && 'PR-' }}${{ github.event.pull_request.number || github.ref_name }}
jobs:
build-cuda13:
name: Build CUDA 13 wheel for ${{ matrix.arch }}
runs-on:
- runs-on=${{ github.run_id }}
- runner=${{ matrix.runner }}
- tag=cuda13-build-cuda13-${{ matrix.arch }}
strategy:
fail-fast: false
matrix:
include:
- arch: aarch64
runner: linux-arm64-cpu
image_repo: xgb-ci.gpu_build_cuda13_rockylinux8_aarch64
- arch: x86_64
runner: linux-amd64-cpu
image_repo: xgb-ci.gpu_build_cuda13_rockylinux8
steps:
# Restart Docker daemon so that it recognizes the ephemeral disks
- run: sudo systemctl restart docker
- uses: actions/checkout@v4
with:
submodules: "true"
- name: Log into Docker registry (AWS ECR)
run: bash ops/pipeline/login-docker-registry.sh
- run: |
bash ops/pipeline/build-cuda13.sh ${{ matrix.image_repo }} ${{ matrix.arch }}
- name: Stash files
run: |
python3 ops/pipeline/manage-artifacts.py upload \
--s3-bucket ${{ env.RUNS_ON_S3_BUCKET_CACHE }} \
--prefix cache/${{ github.run_id }}/build-cuda13-${{ matrix.arch }} \
build/testxgboost ./xgboost python-package/dist/*.whl
test-cpp-cuda13:
name: Google Test (C++) with CUDA 13, arch ${{ matrix.arch }}
needs: [build-cuda13]
runs-on:
- runs-on=${{ github.run_id }}
- runner=${{ matrix.runner }}
- tag=cuda13-test-cpp-cuda13-${{ matrix.arch }}
strategy:
fail-fast: false
matrix:
include:
- arch: aarch64
runner: linux-arm64-gpu
image_repo: xgb-ci.gpu_build_cuda13_rockylinux8_aarch64
- arch: x86_64
runner: linux-amd64-gpu
image_repo: xgb-ci.gpu_build_cuda13_rockylinux8
steps:
# Restart Docker daemon so that it recognizes the ephemeral disks
- run: sudo systemctl restart docker
- uses: actions/checkout@v4
with:
submodules: "true"
- name: Log into Docker registry (AWS ECR)
run: bash ops/pipeline/login-docker-registry.sh
- name: Unstash gtest
run: |
python3 ops/pipeline/manage-artifacts.py download \
--s3-bucket ${{ env.RUNS_ON_S3_BUCKET_CACHE }} \
--prefix cache/${{ github.run_id }}/build-cuda13-${{ matrix.arch }} \
--dest-dir build \
testxgboost
chmod +x build/testxgboost
- run: |
bash ops/pipeline/test-cpp-cuda13.sh ${{ matrix.image_repo }}
test-python-cuda13:
name: Run Python tests with CUDA 13, arch ${{ matrix.arch }}
needs: [build-cuda13]
runs-on:
- runs-on=${{ github.run_id }}
- runner=${{ matrix.runner }}
- tag=cuda13-test-python-cuda13-${{ matrix.arch }}
strategy:
fail-fast: false
matrix:
include:
- arch: x86_64
runner: linux-amd64-gpu
image_repo: xgb-ci.gpu_build_cuda13_rockylinux8
- arch: aarch64
runner: linux-arm64-gpu
image_repo: xgb-ci.gpu_build_cuda13_rockylinux8_aarch64
steps:
# Restart Docker daemon so that it recognizes the ephemeral disks
- run: sudo systemctl restart docker
- uses: actions/checkout@v4
with:
submodules: "true"
- name: Log into Docker registry (AWS ECR)
run: bash ops/pipeline/login-docker-registry.sh
- name: Unstash Python wheel
run: |
python3 ops/pipeline/manage-artifacts.py download \
--s3-bucket ${{ env.RUNS_ON_S3_BUCKET_CACHE }} \
--prefix cache/${{ github.run_id }}/build-cuda13-${{ matrix.arch }} \
--dest-dir wheelhouse \
*.whl xgboost
mv -v wheelhouse/xgboost .
chmod +x ./xgboost
- name: Run Python tests
run: bash ops/pipeline/test-python-wheel-cuda13.sh ${{ matrix.image_repo }}