diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index dd3ac63..0f6c60b 100644 --- a/.github/workflows/build.yml +++ b/.github/workflows/build.yml @@ -29,8 +29,8 @@ jobs: fail-fast: false matrix: flash-attn-version: ["2.8.3"] - python-version: ["3.10", "3.11", "3.12"] - torch-version: ["2.5.1", "2.6.0", "2.7.1", "2.8.0"] + 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: @@ -121,52 +121,52 @@ jobs: # ######################################################### # Windows # ######################################################### - # build_wheels_windows: - # name: Build Windows - # needs: create_releases - # strategy: - # fail-fast: false - # matrix: - # flash-attn-version: ["2.6.3", "2.7.4", "2.8.2"] - # python-version: ["3.10", "3.11", "3.12"] - # torch-version: ["2.5.1", "2.6.0", "2.7.1", "2.8.0"] - # # https://developer.nvidia.com/cuda-toolkit-archive - # cuda-version: ["12.4.1", "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: + 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_code_build: # name: Build Windows (AWS CodeBuild) @@ -219,8 +219,7 @@ jobs: name: Update Release Notes needs: # [build_wheels_linux, build_wheels_linux_self_hosted, build_wheels_windows] - [build_wheels_linux] - # needs: [build_wheels_windows_code_build] + [build_wheels_linux, build_wheels_windows] if: always() runs-on: ubuntu-latest steps: