Skip to content

Commit 21ea1d2

Browse files
authored
ci: Add PyTorch 2.7.0 build (#1063)
## 📌 Description Add a build for PyTorch 2.7.0 https://pytorch.org/blog/pytorch-2-7 and also remove 2.4 to keep only the last 3 PyTorch releases #835 ## 🔍 Related Issues As there is no official release of FlashInfer for PyTorch 2.7 yet, I'm currently building FlashInfer from [source on vLLM](https://github.com/vllm-project/vllm/blob/main/docker/Dockerfile#L258-L261) and that adds to the build time there. --- ## Reviewer Notes PyTorch 2.7 also moves to CUDA 12.8 build and removes 12.4 (and 12.1) Signed-off-by: Huy Do <[email protected]>
1 parent 5035652 commit 21ea1d2

File tree

1 file changed

+16
-7
lines changed

1 file changed

+16
-7
lines changed

.github/workflows/release_wheel.yml

Lines changed: 16 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -25,19 +25,28 @@ jobs:
2525
strategy:
2626
fail-fast: false
2727
matrix:
28-
cuda: ["11.8", "12.1", "12.4", "12.6"]
29-
torch: ["2.4", "2.5", "2.6"]
28+
cuda: ["11.8", "12.1", "12.4", "12.6", "12.8"]
29+
torch: ["2.5", "2.6", "2.7"]
3030
exclude: # We use release_wheel_sglang.yml for faster release and verification. If everything is okay, then we trigger release_wheel.yml. This combination (cuda 12.4 or 11.8 + torch 2.5) is already handled in release_wheel_sglang.yml
31+
- cuda: "11.8"
32+
torch: "2.5"
3133
- cuda: "12.4"
3234
torch: "2.5"
33-
- cuda: "11.8"
35+
# 2.5 supports 11.8, 12.1, and 12.4
36+
- cuda: "12.6"
37+
torch: "2.5"
38+
- cuda: "12.8"
3439
torch: "2.5"
40+
# 2.6 supports 11.8, 12.4, and 12.6
3541
- cuda: "12.1"
3642
torch: "2.6"
37-
- cuda: "12.6"
38-
torch: "2.4"
39-
- cuda: "12.6"
40-
torch: "2.5"
43+
- cuda: "12.8"
44+
torch: "2.6"
45+
# 2.7 supports 11.8, 12.6, and 12.8
46+
- cuda: "12.1"
47+
torch: "2.7"
48+
- cuda: "12.4"
49+
torch: "2.7"
4150

4251

4352
runs-on: [self-hosted]

0 commit comments

Comments
 (0)