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

add aws codebuild

Browse Source
This commit is contained in:
Junya Morioka
2025-08-15 00:49:13 +09:00
parent 8ba7f486e8
commit d7dc50f3c4
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/build_windows_code_build.yml
+3 -2
View File
@@ -30,8 +30,8 @@ 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: 3 NVCC_THREADS: 4
steps: steps:
- uses: actions/checkout@v4 - uses: actions/checkout@v4
@@ -74,6 +74,7 @@ jobs:
- name: Build wheels - name: Build wheels
shell: pwsh shell: pwsh
timeout-minutes: 129600
run: | run: |
.\build_windows.ps1 -FlashAttnVersion "${{ inputs.flash-attn-version }}" -PythonVersion "${{ inputs.python-version }}" -TorchVersion "${{ inputs.torch-version }}" -CudaVersion "${{ inputs.cuda-version }}" .\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 } $wheelName = Get-ChildItem -Path "flash-attention\dist\*.whl" | Select-Object -First 1 | ForEach-Object { $_.Name }