diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index 2c4caae..2eeb8f1 100644 --- a/.github/workflows/build.yml +++ b/.github/workflows/build.yml @@ -25,7 +25,7 @@ jobs: body: | | Flash-Attention | Python | PyTorch | CUDA | |-----------------|--------|---------|------| - | 2.7.3 | 3.10, 3.11, 3.12 | 2.0.1, 2.1.2, 2.2.2, 2.3.1, 2.4.1, 2.5.1 | 11.8.0, 12.1.1, 12.4.1 | + | 2.6.3, 2.7.4.post1 | 3.10, 3.11, 3.12 | 2.0.1, 2.1.2, 2.2.2, 2.3.1, 2.4.1, 2.5.1, 2.6.0 | 12.4.1, 12.6.3 | build_wheels: name: Build wheels and Upload @@ -34,10 +34,10 @@ jobs: strategy: fail-fast: false matrix: - flash-attn-version: ["2.7.3"] + flash-attn-version: ["2.6.3", "2.7.4.post1"] python-version: ["3.10", "3.11", "3.12"] - torch-version: ["2.0.1", "2.1.2", "2.2.2", "2.3.1", "2.4.1", "2.5.1"] - cuda-version: ["11.8.0", "12.1.1", "12.4.1"] + torch-version: ["2.0.1", "2.1.2", "2.2.2", "2.3.1", "2.4.1", "2.5.1", "2.6.0"] + cuda-version: ["12.4.1", "12.6.3"] exclude: # torch < 2.2 does not support Python 3.12 - python-version: "3.12" @@ -94,6 +94,7 @@ jobs: '2.3': [118, 121], \ '2.4': [118, 121, 124], \ '2.5': [118, 121, 124], \ + '2.6': [118, 124, 126], \ }; \ target_cuda_versions = support_cuda_versions[env['MATRIX_TORCH_VERSION']]; \ cuda_version = int(env['MATRIX_CUDA_VERSION']); \