add aws codebuild

.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 }