Skip to content

Commit

Permalink
ci: add torch 12.4 to the matrix configuration (#398)
Browse files Browse the repository at this point in the history
- Remove torch 2.1 (we will not support torch 2.1 from v0.1.2 on).
- Add pytorch 2.4
- Add cuda 12.4 (for torch 2.4+)
  • Loading branch information
yzh119 authored Jul 26, 2024
1 parent 701c813 commit de16915
Showing 1 changed file with 8 additions and 2 deletions.
10 changes: 8 additions & 2 deletions .github/workflows/release_wheel.yml
Original file line number Diff line number Diff line change
Expand Up @@ -26,8 +26,14 @@ jobs:
fail-fast: false
matrix:
python: ["3.8", "3.9", "3.10", "3.11"]
cuda: ["11.8", "12.1"]
torch: ["2.1", "2.2", "2.3"]
cuda: ["11.8", "12.1", "12.4"]
torch: ["2.2", "2.3", "2.4"]
exclude: # for cuda 12.4, we only support torch 2.4+
- cuda: "12.4"
torch: "2.2"
- cuda: "12.4"
torch: "2.3"

runs-on: [self-hosted]
steps:
- uses: actions/checkout@v4
Expand Down

0 comments on commit de16915

Please sign in to comment.