yindo/flash-attention-prebuild-wheels-rocm
1
0
Fork 0
mirror of https://github.com/BillyOutlast/flash-attention-prebuild-wheels-rocm.git synced 2026-07-01 01:37:53 -04:00
Code Issues Packages Projects Releases 2 Wiki Activity

fix aws codebuild

Browse Source
This commit is contained in:
Junya Morioka
2025-08-15 05:16:14 +09:00
parent 74600ce68f
commit 869d45aaad
2 changed files with 1 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/build_windows.yml
-13
View File
@@ -91,17 +91,4 @@ jobs:
run: | run: |
$tag_name = "${env:GITHUB_REF}".Replace("refs/tags/", "") $tag_name = "${env:GITHUB_REF}".Replace("refs/tags/", "")
$wheel_path = "flash-attention/dist/$env:wheel_name" $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 gh release upload "$tag_name" "$wheel_path" --clobber
.github/workflows/build_windows_code_build.yml
+1 -14
View File
@@ -30,7 +30,7 @@ jobs:
runs-on: codebuild-flash-attention-pre-build-wheel-windows-${{ github.run_id }}-${{ github.run_attempt }} runs-on: codebuild-flash-attention-pre-build-wheel-windows-${{ github.run_id }}-${{ github.run_attempt }}
# Large Instance # Large Instance
env: env:
MAX_JOBS: 4 MAX_JOBS: 6
NVCC_THREADS: 2 NVCC_THREADS: 2
steps: steps:
- uses: actions/checkout@v4 - uses: actions/checkout@v4
@@ -93,17 +93,4 @@ jobs:
run: | run: |
$tag_name = "${env:GITHUB_REF}".Replace("refs/tags/", "") $tag_name = "${env:GITHUB_REF}".Replace("refs/tags/", "")
$wheel_path = "flash-attention/dist/$env:wheel_name" $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 gh release upload "$tag_name" "$wheel_path" --clobber