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:37: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" name: "Test Windows build"
on: on:
workflow_dispatch: workflow_dispatch:
@@ -65,8 +67,7 @@ jobs:
- name: Install build dependencies - name: Install build dependencies
run: | run: |
choco install -y ninja pip install -U pip setuptools==75.8.0 wheel setuptools packaging psutil ninja
pip install -U pip setuptools==75.8.0 wheel setuptools packaging psutil
shell: pwsh shell: pwsh
- name: Install PyTorch ${{ matrix.torch-version }}+cu${{ matrix.cuda-version }} - name: Install PyTorch ${{ matrix.torch-version }}+cu${{ matrix.cuda-version }}
@@ -99,9 +100,9 @@ jobs:
$env:DISTUTILS_USE_SDK = 1 $env:DISTUTILS_USE_SDK = 1
$env:BUILD_TARGET = "cuda" $env:BUILD_TARGET = "cuda"
$env:CUDA_HOME = $env:CUDA_PATH $env:CUDA_PATH = $env:CONDA_PREFIX
$env:CUDA_ROOT = $env:CUDA_PATH $env:CUDA_HOME = $env:CONDA_PREFIX
$env:PATH = "$env:CUDA_PATH\bin;$env:PATH"
$env:MAX_JOBS = "2" $env:MAX_JOBS = "2"
$env:NVCC_THREADS = "2" $env:NVCC_THREADS = "2"
$env:FLASH_ATTENTION_FORCE_BUILD = "TRUE" $env:FLASH_ATTENTION_FORCE_BUILD = "TRUE"