diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index bbd41a8..58d948d 100644 --- a/.github/workflows/build.yml +++ b/.github/workflows/build.yml @@ -30,7 +30,7 @@ jobs: matrix: flash-attn-version: ["2.8.3"] python-version: ["3.11", "3.12", "3.13"] - torch-version: ["2.9.0.dev"] + torch-version: ["2.9.0.dev20250909"] # https://developer.nvidia.com/cuda-toolkit-archive cuda-version: ["12.6.3", "12.8.1", "12.9.1"] exclude: @@ -121,52 +121,52 @@ jobs: # ######################################################### # Windows # ######################################################### - # build_wheels_windows: - # name: Build Windows - # needs: create_releases - # strategy: - # fail-fast: false - # matrix: - # flash-attn-version: ["2.8.2"] - # python-version: ["3.13"] - # torch-version: ["2.6.0", "2.7.1", "2.8.0"] - # # https://developer.nvidia.com/cuda-toolkit-archive - # cuda-version: ["12.4.1", "12.6.3", "12.8.1", "12.9.1"] - # exclude: - # # torch < 2.2 does not support Python 3.12 - # - python-version: "3.12" - # torch-version: "2.0.1" - # - python-version: "3.12" - # torch-version: "2.1.2" - # # torch 2.0.1 does not support CUDA 12.x - # - torch-version: "2.0.1" - # cuda-version: "12.1.1" - # - torch-version: "2.0.1" - # cuda-version: "12.4.1" - # - torch-version: "2.0.1" - # cuda-version: "12.6.3" - # - torch-version: "2.0.1" - # cuda-version: "12.8.1" - # # torch 2.6.0 does not support CUDA 12.1 - # - torch-version: "2.6.0" - # cuda-version: "12.1.1" - # # torch 2.7.0 does not support CUDA 12.4 - # - torch-version: "2.7.0" - # cuda-version: "12.4.1" - # # torch < 2.8 does not support CUDA 12.9 - # - torch-version: "2.5.1" - # cuda-version: "12.9.1" - # - torch-version: "2.6.3" - # cuda-version: "12.9.1" - # - torch-version: "2.7.1" - # cuda-version: "12.9.1" - # uses: ./.github/workflows/build_windows.yml - # with: - # flash-attn-version: ${{ matrix.flash-attn-version }} - # python-version: ${{ matrix.python-version }} - # torch-version: ${{ matrix.torch-version }} - # cuda-version: ${{ matrix.cuda-version }} - # secrets: inherit + build_wheels_windows: + name: Build Windows + needs: create_releases + strategy: + fail-fast: false + matrix: + flash-attn-version: ["2.8.3"] + python-version: ["3.11", "3.12", "3.13"] + torch-version: ["2.9.0.dev20250909"] + # https://developer.nvidia.com/cuda-toolkit-archive + cuda-version: ["12.6.3", "12.8.1"] + exclude: + # torch < 2.2 does not support Python 3.12 + - python-version: "3.12" + torch-version: "2.0.1" + - python-version: "3.12" + torch-version: "2.1.2" + # torch 2.0.1 does not support CUDA 12.x + - torch-version: "2.0.1" + cuda-version: "12.1.1" + - torch-version: "2.0.1" + cuda-version: "12.4.1" + - torch-version: "2.0.1" + cuda-version: "12.6.3" + - torch-version: "2.0.1" + cuda-version: "12.8.1" + # torch 2.6.0 does not support CUDA 12.1 + - torch-version: "2.6.0" + cuda-version: "12.1.1" + # torch 2.7.0 does not support CUDA 12.4 + - torch-version: "2.7.0" + cuda-version: "12.4.1" + # torch < 2.8 does not support CUDA 12.9 + - torch-version: "2.5.1" + cuda-version: "12.9.1" + - torch-version: "2.6.3" + cuda-version: "12.9.1" + - torch-version: "2.7.1" + cuda-version: "12.9.1" + uses: ./.github/workflows/build_windows.yml + with: + flash-attn-version: ${{ matrix.flash-attn-version }} + python-version: ${{ matrix.python-version }} + torch-version: ${{ matrix.torch-version }} + cuda-version: ${{ matrix.cuda-version }} + secrets: inherit # build_wheels_windows_code_build: # name: Build Windows (AWS CodeBuild) @@ -219,7 +219,7 @@ jobs: name: Update Release Notes needs: # [build_wheels_linux, build_wheels_linux_self_hosted, build_wheels_windows] - [build_wheels_linux] + [build_wheels_linux, build_wheels_windows] permissions: contents: write if: always()