Skip to content

Simplify GitHub Actions CI: abstract CUDA versions to cu11/cu12 - #95

Merged
teddykoker merged 1 commit into
teddykoker:mainfrom
LouisJalouzot:ci/simplify-cuda-matrix
Mar 3, 2026
Merged

teddykoker merged 1 commit into
teddykoker:mainfrom
LouisJalouzot:ci/simplify-cuda-matrix

Conversation

@LouisJalouzot

Copy link
Copy Markdown
Contributor

Changes

  • Replace custom install.sh script with Jimver/cuda-toolkit action
  • Abstract CUDA versions to cu11/cu12 (system CUDA independent of PyTorch wheel index)
  • Add PyTorch 2.7.0 and 2.8.0 support
  • Add Python 3.13 support
  • Expand GPU targets: added sm_89 (Ada) and sm_90+ (Hopper)
  • Bump action versions to v4/v5

Building the wheels works (c.f. this release on my fork) and I could try that torchsort-0.1.10+pt27cu12-cp313-cp313-linux_x86_64.whl works on GPU in practice.

@teddykoker

Copy link
Copy Markdown
Owner

LGTM, thanks @LouisJalouzot!

@teddykoker
teddykoker merged commit dda8a16 into teddykoker:main Mar 3, 2026
57 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants