diff --git a/.github/workflows/build_windows.yml b/.github/workflows/build_windows.yml index ad21a7b..b4a53fe 100644 --- a/.github/workflows/build_windows.yml +++ b/.github/workflows/build_windows.yml @@ -91,17 +91,4 @@ jobs: run: | $tag_name = "${env:GITHUB_REF}".Replace("refs/tags/", "") $wheel_path = "flash-attention/dist/$env:wheel_name" - - # Check if the file exists - if (-not (Test-Path $wheel_path)) { - $tag_name = "${env:GITHUB_REF}".Replace("refs/tags/", "") - $wheel_path = "flash-attention/dist/$env:wheel_name" - - # Check if the file exists - if (-not (Test-Path $wheel_path)) { - Write-Host "Error: Wheel file not found at $wheel_path" - exit 1 - } - - # Upload the release asset using GitHub CLI gh release upload "$tag_name" "$wheel_path" --clobber diff --git a/.github/workflows/build_windows_code_build.yml b/.github/workflows/build_windows_code_build.yml index b50b7f9..116a173 100644 --- a/.github/workflows/build_windows_code_build.yml +++ b/.github/workflows/build_windows_code_build.yml @@ -30,7 +30,7 @@ jobs: runs-on: codebuild-flash-attention-pre-build-wheel-windows-${{ github.run_id }}-${{ github.run_attempt }} # Large Instance env: - MAX_JOBS: 4 + MAX_JOBS: 6 NVCC_THREADS: 2 steps: - uses: actions/checkout@v4 @@ -93,17 +93,4 @@ jobs: run: | $tag_name = "${env:GITHUB_REF}".Replace("refs/tags/", "") $wheel_path = "flash-attention/dist/$env:wheel_name" - - # Check if the file exists - if (-not (Test-Path $wheel_path)) { - $tag_name = "${env:GITHUB_REF}".Replace("refs/tags/", "") - $wheel_path = "flash-attention/dist/$env:wheel_name" - - # Check if the file exists - if (-not (Test-Path $wheel_path)) { - Write-Host "Error: Wheel file not found at $wheel_path" - exit 1 - } - - # Upload the release asset using GitHub CLI gh release upload "$tag_name" "$wheel_path" --clobber