Skip to content

Trigger workflow

Trigger workflow #2

Workflow file for this run

name: Unit tests
on:
workflow_dispatch:
push:
branches: [gpu-ci]
#pull_request:
# branches: [main, multi-backend-refactor]
jobs:
run-tests:
runs-on:
group: CUDA-Linux-x64
needs: build-cuda
strategy:
fail-fast: false
matrix:
torch_version: [
# "2.0.1",
# "2.1.2",
# "2.2.2",
"2.3.1",
"2.4.1"
]
python_version:
["3.8", "3.9", "3.10", "3.11", "3.12"]
steps:
- name: Checkout repo
uses: actions/checkout@v4
- name: Download build artifact
uses: actions/download-artifact@v4
with:
pattern: shared_cuda*
merge-multiple: true
path: bitsandbytes/
- name: Show directory structure
run: ls -lR
- name: Show GPU information
run: nvidia-smi
- name: Set up Python ${{ matrix.python_version }}
uses: actions/setup-python@v5
with:
python-version: ${{ matrix.python_version }}
- name: Install testing dependencies
run: |
pip install -r requirements-ci.txt
pip install -e .
pip install torch==${{ matrix.torch_version }}
- name: Show installed packages
run: pip freeze
- name: Run tests
run: pytest
build-cuda:
runs-on: ubuntu-latest
strategy:
fail-fast: true
matrix:
cuda_version:
["11.8.0", "12.1.1", "12.4.1"]
steps:
- name: Checkout repo
uses: actions/checkout@v4
- name: Build CUDA backend
run: bash .github/scripts/build-cuda.sh
env:
build_os: ubuntu
build_arch: x86_64
cuda_version: ${{ matrix.cuda_version }}
- name: Upload build artifact
uses: actions/upload-artifact@v4
with:
name: shared_cuda${{ matrix.cuda_version }}
path: output/*
retention-days: 7