From fccc712daeb3f8811ca0beaba6769b99a57c94ba Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Wed, 11 Mar 2026 20:09:03 -0400 Subject: [PATCH] builds for py312-cu128-torch2.9.1 (#3489) --- .github/workflows/base.yml | 8 ++++++++ .github/workflows/main.yml | 12 ++++++++++++ 2 files changed, 20 insertions(+) diff --git a/.github/workflows/base.yml b/.github/workflows/base.yml index 10326fba9..29842bcb5 100644 --- a/.github/workflows/base.yml +++ b/.github/workflows/base.yml @@ -173,6 +173,14 @@ jobs: torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" dockerfile: "Dockerfile-uv-base" platforms: "linux/amd64,linux/arm64" + - cuda: "128" + cuda_version: 12.8.1 + cudnn_version: "" + python_version: "3.12" + pytorch: 2.9.1 + torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" + dockerfile: "Dockerfile-uv-base" + platforms: "linux/amd64,linux/arm64" - cuda: "128" cuda_version: 12.8.1 cudnn_version: "" diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml index aca9b1dd3..4f772381f 100644 --- a/.github/workflows/main.yml +++ b/.github/workflows/main.yml @@ -110,6 +110,12 @@ jobs: pytorch: 2.9.1 axolotl_extras: platforms: "linux/amd64,linux/arm64" + - cuda: 128 + cuda_version: 12.8.1 + python_version: "3.12" + pytorch: 2.9.1 + axolotl_extras: + platforms: "linux/amd64,linux/arm64" is_latest: true - cuda: 128 cuda_version: 12.8.1 @@ -193,6 +199,12 @@ jobs: python_version: "3.11" pytorch: 2.9.1 axolotl_extras: + platforms: "linux/amd64,linux/arm64" + - cuda: 128 + cuda_version: 12.8.1 + python_version: "3.12" + pytorch: 2.9.1 + axolotl_extras: is_latest: true platforms: "linux/amd64,linux/arm64" - cuda: 128