From 3bed0f11cacf56762b13b096bb5d6bc9fa2264bf Mon Sep 17 00:00:00 2001 From: Junya Morioka Date: Wed, 11 Jun 2025 23:35:23 +0900 Subject: [PATCH] [WIP] add aws code build --- .github/workflows/test-code-build.yml | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/.github/workflows/test-code-build.yml b/.github/workflows/test-code-build.yml index 6e0765d..e1a2b4a 100644 --- a/.github/workflows/test-code-build.yml +++ b/.github/workflows/test-code-build.yml @@ -8,8 +8,8 @@ jobs: # runs-on: windows-latest runs-on: codebuild-flash-attention-pre-build-wheel-windows-${{ github.run_id }}-${{ github.run_attempt }} env: - MAX_JOBS: 4 - NVCC_THREADS: 4 + MAX_JOBS: 2 + NVCC_THREADS: 2 steps: - uses: actions/checkout@v4 - name: Enable Git long paths @@ -22,7 +22,7 @@ jobs: - uses: Jimver/cuda-toolkit@v0.2.24 with: - cuda: "12.4.1" + cuda: "12.8.1" method: "network" use-github-cache: false use-local-cache: false @@ -42,7 +42,7 @@ jobs: - name: Build wheels shell: pwsh run: | - .\build_windows.ps1 -FlashAttnVersion "2.5.9" -PythonVersion "3.11" -TorchVersion "2.6.0" -CudaVersion "12.4.1" + .\build_windows.ps1 -FlashAttnVersion "2.7.4" -PythonVersion "3.11" -TorchVersion "2.7.1" -CudaVersion "12.8.1" $wheelName = Get-ChildItem -Path "flash-attention\dist\*.whl" | Select-Object -First 1 | ForEach-Object { $_.Name } echo "wheel_name=$wheelName" >> $env:GITHUB_ENV