yindo/flash-attention-prebuild-wheels-rocm
1
0
Fork 0
mirror of https://github.com/BillyOutlast/flash-attention-prebuild-wheels-rocm.git synced 2026-06-30 23:57:53 -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 01:25:28 +09:00
parent 6afe710ecd
commit e78c78d5e6
3 changed files with 18 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/build_windows.yml
+9 -3
View File
@@ -28,9 +28,6 @@ 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
@@ -73,7 +70,16 @@ 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
+6
View File
@@ -76,6 +76,12 @@ jobs:
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
build_windows.ps1
+3
View File
@@ -61,6 +61,9 @@ $env:BUILD_TARGET = "cuda"
# Use environment variables from workflow if available, otherwise use defaults
if (-not $env:MAX_JOBS) { $env:MAX_JOBS = "2" }
if (-not $env:NVCC_THREADS) { $env:NVCC_THREADS = "2" }
Write-Host "Environment variables:"
Write-Host " MAX_JOBS: $env:MAX_JOBS"
Write-Host " NVCC_THREADS: $env:NVCC_THREADS"
$env:FLASH_ATTENTION_FORCE_BUILD = "TRUE"
$env:NVCC_FLAGS = "-w --disable-warnings"
$env:CXXFLAGS = "/w"