Skip to content

Reduce CUDA wheel size for PyPI publishing #6

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 3 commits into from
Jun 25, 2026
Merged

Reduce CUDA wheel size for PyPI publishing #6

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
de4ae9d
Reduce CUDA wheel size
andimarafioti Jun 25, 2026
bc6601d
Limit validation wheels to CUDA 12.8
andimarafioti Jun 25, 2026
404699d
Build CUDA 13 validation wheels
andimarafioti Jun 25, 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
4 changes: 2 additions & 2 deletions .github/workflows/publish.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -27,14 +27,14 @@ jobs:
cuda_repo_arch: x86_64
cuda_package_suffix: "12-8"
cuda_home: /usr/local/cuda-12.8
cuda_architectures: 75-virtual;80-virtual;86-real;89-real;90-real;100-real;120-real;120-virtual
cuda_architectures: 75-virtual;86-real;90-real;120-real;120-virtual
build_jobs: "2"
- arch: aarch64
runner: ubuntu-24.04-arm
cuda_repo_arch: sbsa
cuda_package_suffix: "12-8"
cuda_home: /usr/local/cuda-12.8
cuda_architectures: 75-virtual;80-virtual;86-real;89-real;90-real;100-real;120-real;120-virtual
cuda_architectures: 75-virtual;86-real;90-real;120-real;120-virtual
build_jobs: "2"
env:
QWENTTS_CPP_BACKEND: cuda
Expand Down
8 changes: 4 additions & 4 deletions .github/workflows/wheels.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ jobs:
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
cuda_architectures: 75-virtual;86-real;90-real;120-real;120-virtual
build_jobs: "2"
- arch: aarch64
runner: ubuntu-24.04-arm
Expand All @@ -42,7 +42,7 @@ jobs:
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
cuda_architectures: 75-virtual;86-real;90-real;120-real;120-virtual
build_jobs: "2"
- arch: x86_64
runner: ubuntu-24.04
Expand All @@ -51,7 +51,7 @@ jobs:
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
cuda_architectures: 75-virtual;80-real;86-real;90-real;121-real
build_jobs: "2"
- arch: aarch64
runner: ubuntu-24.04-arm
Expand All @@ -60,7 +60,7 @@ jobs:
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
cuda_architectures: 75-virtual;80-real;86-real;90-real;121-real
build_jobs: "2"
env:
QWENTTS_CPP_BACKEND: cuda
Expand Down
19 changes: 16 additions & 3 deletions AGENTS.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -12,9 +12,22 @@ The workflow runs on pushed tags that match `v*`, builds the public Linux CUDA
through the configured `pypi` environment.

The public `qwentts-cpp-python` package currently publishes CUDA 12.8 wheels
only. CPU and CUDA 13 wheels are useful validation artifacts, but publishing
multiple backend flavors under the same package name, version, and platform tag
would leave pip with no reliable way to choose the intended runtime.
only. Publishing multiple backend flavors under the same package name, version,
and platform tag would leave pip with no reliable way to choose the intended
runtime.

The PR wheel workflow may still build CPU and CUDA 13 wheels as downloadable
validation artifacts. Those artifacts are not uploaded to PyPI by the publish
workflow.

CUDA release wheels keep native cubins for sm_86, sm_90, and sm_120, plus PTX
fallbacks for sm_75 and the newest supported CUDA architecture. This keeps the
public CUDA 12.8 wheels under PyPI's default per-file upload limit while
preserving broad GPU compatibility. Ada sm_89 GPUs can run the sm_86 cubin
through CUDA's same-major binary compatibility. DGX Spark / GB10 sm_121 uses
PTX fallback in the public CUDA 12.8 wheels. A CUDA 13 build with native sm_121
can avoid that fallback, but it is not the PyPI default because the deployment
server image currently targets CUDA 12.8.

To prepare a release:

Expand Down
Loading