Skip to content

Run x86 CUDA wheels on HF Jobs #3

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
andimarafioti merged 4 commits into from
Jun 24, 2026
Merged

Run x86 CUDA wheels on HF Jobs #3

Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
244c53f
Run CUDA 13 wheels on HF Jobs
andimarafioti Jun 24, 2026
0007323
Run x86 CUDA wheels on HF Jobs
andimarafioti Jun 24, 2026
12b1de8
Remove workflow runner comment
andimarafioti Jun 24, 2026
0d7d519
Remove runner setup docs
andimarafioti Jun 24, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 7 additions & 3 deletions .github/workflows/wheels.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -27,13 +27,14 @@ jobs:
matrix:
include:
- arch: x86_64
runner: ubuntu-24.04
runner: hf-jobs-cpu-performance
cuda_repo_arch: x86_64
cuda_version: "12.8"
cuda_package_suffix: "12-8"
cuda_home: /usr/local/cuda-12.8
wheel_build_tag: 1cu128
cuda_architectures: 75-virtual;80-virtual;86-real;89-real;90-real;100-real;120-real;120-virtual
build_jobs: "16"
- arch: aarch64
runner: ubuntu-24.04-arm
cuda_repo_arch: sbsa
Expand All @@ -42,14 +43,16 @@ jobs:
cuda_home: /usr/local/cuda-12.8
wheel_build_tag: 1cu128
cuda_architectures: 75-virtual;80-virtual;86-real;89-real;90-real;100-real;120-real;120-virtual
build_jobs: "2"
- arch: x86_64
runner: ubuntu-24.04
runner: hf-jobs-cpu-performance
cuda_repo_arch: x86_64
cuda_version: "13.0"
cuda_package_suffix: "13-0"
cuda_home: /usr/local/cuda-13.0
wheel_build_tag: 1cu130
cuda_architectures: 75-virtual;80-virtual;86-real;89-real;90-real;100-real;120-real;120-virtual;121-real;121-virtual
build_jobs: "16"
- arch: aarch64
runner: ubuntu-24.04-arm
cuda_repo_arch: sbsa
Expand All @@ -58,9 +61,10 @@ jobs:
cuda_home: /usr/local/cuda-13.0
wheel_build_tag: 1cu130
cuda_architectures: 75-virtual;80-virtual;86-real;89-real;90-real;100-real;120-real;120-virtual;121-real;121-virtual
build_jobs: "2"
env:
QWENTTS_CPP_BACKEND: cuda
QWENTTS_CPP_BUILD_JOBS: "2"
QWENTTS_CPP_BUILD_JOBS: ${{ matrix.build_jobs }}
QWENTTS_CPP_WHEEL_BUILD_TAG: ${{ matrix.wheel_build_tag }}
QWENTTS_REF: ${{ inputs.qwentts_ref || '4536dcdce27c3764a93a06d6bf64026b124962f5' }}
CUDA_ARCHITECTURES: ${{ matrix.cuda_architectures }}
Expand Down
Loading