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

[WIP] add aws code build

Browse Source
This commit is contained in:
Junya Morioka
2025-06-11 23:35:23 +09:00
parent d5600115fc
commit 3bed0f11ca
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/test-code-build.yml
+4 -4
View File
@@ -8,8 +8,8 @@ jobs:
# runs-on: windows-latest # runs-on: windows-latest
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 }}
env: env:
MAX_JOBS: 4 MAX_JOBS: 2
NVCC_THREADS: 4 NVCC_THREADS: 2
steps: steps:
- uses: actions/checkout@v4 - uses: actions/checkout@v4
- name: Enable Git long paths - name: Enable Git long paths
@@ -22,7 +22,7 @@ jobs:
- uses: Jimver/cuda-toolkit@v0.2.24 - uses: Jimver/cuda-toolkit@v0.2.24
with: with:
cuda: "12.4.1" cuda: "12.8.1"
method: "network" method: "network"
use-github-cache: false use-github-cache: false
use-local-cache: false use-local-cache: false
@@ -42,7 +42,7 @@ jobs:
- name: Build wheels - name: Build wheels
shell: pwsh shell: pwsh
run: | run: |
.\build_windows.ps1 -FlashAttnVersion "2.5.9" -PythonVersion "3.11" -TorchVersion "2.6.0" -CudaVersion "12.4.1" .\build_windows.ps1 -FlashAttnVersion "2.7.4" -PythonVersion "3.11" -TorchVersion "2.7.1" -CudaVersion "12.8.1"
$wheelName = Get-ChildItem -Path "flash-attention\dist\*.whl" | Select-Object -First 1 | ForEach-Object { $_.Name } $wheelName = Get-ChildItem -Path "flash-attention\dist\*.whl" | Select-Object -First 1 | ForEach-Object { $_.Name }
echo "wheel_name=$wheelName" >> $env:GITHUB_ENV echo "wheel_name=$wheelName" >> $env:GITHUB_ENV