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

fix: MAX_JOBS,NVCC_THREADS in self-hosted

Browse Source
This commit is contained in:
Junya Morioka
2025-06-16 02:14:42 +09:00
parent 790352b8ed
commit 826618d534
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/build.yml
+2 -2
View File
@@ -253,8 +253,8 @@ jobs:
timeout-minutes: 1200
shell: bash
env:
MAX_JOBS: 4
NVCC_THREADS: 4
MAX_JOBS: 2
NVCC_THREADS: 2
run: |
chmod +x build_linux.sh
./build_linux.sh ${{ matrix.flash-attn-version }} ${{ matrix.python-version }} ${{ matrix.torch-version }} ${{ matrix.cuda-version }}