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:27:54 -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-08-10 08:33:29 +09:00
parent 94d53f9bea
commit 16886f8930
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/test-code-build.yml
+2 -2
View File
@@ -64,8 +64,8 @@ jobs:
timeout-minutes: 1200 timeout-minutes: 1200
shell: pwsh shell: pwsh
env: env:
MAX_JOBS: 2 MAX_JOBS: 1
NVCC_THREADS: 2 NVCC_THREADS: 1
run: | run: |
.\build_windows.ps1 -FlashAttnVersion "2.7.4" -PythonVersion "3.11" -TorchVersion "2.7.1" -CudaVersion "12.8.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 }