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

fix aws codebuild

Browse Source
This commit is contained in:
Junya Morioka
2025-08-15 02:13:08 +09:00
parent 98dfdf7b55
commit 0b49ce54e3
2 changed files with 7 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/build_windows.yml
+3 -9
View File
@@ -28,6 +28,9 @@ jobs:
build_windows_wheels:
name: Build wheels and Upload (Windows x86_64, GitHub hosted runner, ${{ inputs.flash-attn-version }}, ${{ inputs.python-version }}, ${{ inputs.torch-version }}, ${{ inputs.cuda-version }})
runs-on: windows-2022
env:
MAX_JOBS: 2
NVCC_THREADS: 2
steps:
- uses: actions/checkout@v4
@@ -70,16 +73,7 @@ jobs:
- name: Build wheels
shell: pwsh
env:
MAX_JOBS: 2
NVCC_THREADS: 2
run: |
$env:MAX_JOBS = "${{ env.MAX_JOBS }}"
$env:NVCC_THREADS = "${{ env.NVCC_THREADS }}"
$env:FLASH_ATTENTION_FORCE_BUILD = "TRUE"
$env:NVCC_FLAGS = "-w --disable-warnings"
$env:CXXFLAGS = "/w"
$env:CFLAGS = "/w"
.\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 }
echo "wheel_name=$wheelName" >> $env:GITHUB_ENV
.github/workflows/build_windows_code_build.yml
+4 -10
View File
@@ -28,6 +28,10 @@ jobs:
name: Build wheels and Upload (Windows x86_64, AWS CodeBuild, ${{ inputs.flash-attn-version }}, ${{ inputs.python-version }}, ${{ inputs.torch-version }}, ${{ inputs.cuda-version }})
timeout-minutes: 129600
runs-on: codebuild-flash-attention-pre-build-wheel-windows-${{ github.run_id }}-${{ github.run_attempt }}
# Large Instance
env:
MAX_JOBS: 4
NVCC_THREADS: 3
steps:
- uses: actions/checkout@v4
@@ -71,17 +75,7 @@ jobs:
- name: Build wheels
shell: pwsh
timeout-minutes: 129600
# Large Instance
env:
MAX_JOBS: 6
NVCC_THREADS: 4
run: |
$env:MAX_JOBS = "${{ env.MAX_JOBS }}"
$env:NVCC_THREADS = "${{ env.NVCC_THREADS }}"
$env:FLASH_ATTENTION_FORCE_BUILD = "TRUE"
$env:NVCC_FLAGS = "-w --disable-warnings"
$env:CXXFLAGS = "/w"
$env:CFLAGS = "/w"
.\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 }
echo "wheel_name=$wheelName" >> $env:GITHUB_ENV