mirror of
https://github.com/BillyOutlast/flash-attention-prebuild-wheels-rocm.git
synced 2026-07-01 01:37:53 -04:00
[WIP] add workflow for windows
This commit is contained in:
@@ -54,7 +54,6 @@ jobs:
|
||||
with:
|
||||
cuda: ${{ matrix.cuda-version }}
|
||||
method: "network"
|
||||
sub-packages: '["nvcc", "cudart", "cublas", "cufft", "curand", "cusolver", "cusparse", "visual_studio_integration"]'
|
||||
|
||||
- name: Set CUDA and PyTorch versions
|
||||
run: |
|
||||
@@ -100,9 +99,6 @@ jobs:
|
||||
$env:DISTUTILS_USE_SDK = 1
|
||||
$env:BUILD_TARGET = "cuda"
|
||||
|
||||
$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"
|
||||
|
||||
Reference in New Issue
Block a user