diff --git a/.github/workflows/build_linux_self_host.yml b/.github/workflows/build_linux_self_host.yml index 18ae01c..6b33f7b 100644 --- a/.github/workflows/build_linux_self_host.yml +++ b/.github/workflows/build_linux_self_host.yml @@ -36,7 +36,7 @@ jobs: env: DEBIAN_FRONTEND: noninteractive TERM: xterm-256color - timeout-minutes: 10000 + timeout-minutes: 2160 steps: - name: Install tools shell: bash @@ -101,7 +101,7 @@ jobs: export LD_LIBRARY_PATH=/usr/local/nvidia/lib64:/usr/local/cuda/lib64:$LD_LIBRARY_PATH - name: Build wheels - timeout-minutes: 10000 + timeout-minutes: 2160 id: build_wheels shell: bash env: diff --git a/.github/workflows/build_windows_code_build.yml b/.github/workflows/build_windows_code_build.yml index bb63566..aecd662 100644 --- a/.github/workflows/build_windows_code_build.yml +++ b/.github/workflows/build_windows_code_build.yml @@ -26,7 +26,7 @@ on: jobs: build_wheels: 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 + timeout-minutes: 2160 runs-on: codebuild-flash-attention-pre-build-wheel-windows-${{ github.run_id }}-${{ github.run_attempt }} # Large Instance env: @@ -74,7 +74,7 @@ jobs: - name: Build wheels shell: pwsh - timeout-minutes: 129600 + timeout-minutes: 2160 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 }