From d7dc50f3c4a4e97decf7f005164cf2cabc2cfb70 Mon Sep 17 00:00:00 2001
From: Junya Morioka <mjun@mjunya.com>
Date: Fri, 15 Aug 2025 00:49:13 +0900
Subject: [PATCH] add aws codebuild

---
 .github/workflows/build_windows_code_build.yml | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/build_windows_code_build.yml b/.github/workflows/build_windows_code_build.yml
index 46b3d96..1de46b5 100644
--- a/.github/workflows/build_windows_code_build.yml
+++ b/.github/workflows/build_windows_code_build.yml
@@ -30,8 +30,8 @@ jobs:
     runs-on: codebuild-flash-attention-pre-build-wheel-windows-${{ github.run_id }}-${{ github.run_attempt }}
     # Large Instance
     env:
-      MAX_JOBS: 4
-      NVCC_THREADS: 3
+      MAX_JOBS: 6
+      NVCC_THREADS: 4
     steps:
       - uses: actions/checkout@v4
 
@@ -74,6 +74,7 @@ jobs:
 
       - name: Build wheels
         shell: pwsh
+        timeout-minutes: 129600
         run: |
           .\build_windows.ps1 -FlashAttnVersion "${{ inputs.flash-attn-version }}" -PythonVersion "${{ inputs.python-version }}" -TorchVersion "${{ inputs.torch-version }}" -CudaVersion "${{ inputs.cuda-version }}"
           $wheelName = Get-ChildItem -Path "flash-attention\dist\*.whl" | Select-Object -First 1 | ForEach-Object { $_.Name }