add aws codebuild

.github/workflows/build_windows_code_build.yml

@@ -28,10 +28,6 @@ jobs:
     name: Build wheels and Upload (Windows x86_64, AWS CodeBuild, ${{ inputs.flash-attn-version }}, ${{ inputs.python-version }}, ${{ inputs.torch-version }}, ${{ inputs.cuda-version }})
     timeout-minutes: 129600
     runs-on: codebuild-flash-attention-pre-build-wheel-windows-${{ github.run_id }}-${{ github.run_attempt }}
-    # Large Instance
-    env:
-      MAX_JOBS: 6
-      NVCC_THREADS: 4
     steps:
       - uses: actions/checkout@v4
 
@@ -75,6 +71,10 @@ jobs:
       - name: Build wheels
         shell: pwsh
         timeout-minutes: 129600
+        # Large Instance
+        env:
+          MAX_JOBS: 6
+          NVCC_THREADS: 4
         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 }