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

[WIP] add workflow for windows

Browse Source
This commit is contained in:
Junya Morioka
2025-05-24 06:01:41 +09:00
parent 99e6a665ac
commit 8a294a2681
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/test-windows-build.yml
+6 -5
View File
@@ -1,3 +1,5 @@
# Ref: https://github.com/Foul-Tarnished/flash-attention/blob/main/.github/workflows/build-wheels.yml
name: "Test Windows build"
on:
workflow_dispatch:
@@ -65,8 +67,7 @@ jobs:
- name: Install build dependencies
run: |
choco install -y ninja
pip install -U pip setuptools==75.8.0 wheel setuptools packaging psutil
pip install -U pip setuptools==75.8.0 wheel setuptools packaging psutil ninja
shell: pwsh
- name: Install PyTorch ${{ matrix.torch-version }}+cu${{ matrix.cuda-version }}
@@ -99,9 +100,9 @@ jobs:
$env:DISTUTILS_USE_SDK = 1
$env:BUILD_TARGET = "cuda"
$env:CUDA_HOME = $env:CUDA_PATH
$env:CUDA_ROOT = $env:CUDA_PATH
$env:PATH = "$env:CUDA_PATH\bin;$env:PATH"
$env:CUDA_PATH = $env:CONDA_PREFIX
$env:CUDA_HOME = $env:CONDA_PREFIX
$env:MAX_JOBS = "2"
$env:NVCC_THREADS = "2"
$env:FLASH_ATTENTION_FORCE_BUILD = "TRUE"