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 set timeout-minutes, update MAX_JOBS

Browse Source
This commit is contained in:
Junya Morioka
2025-04-27 23:00:33 +09:00
parent 6737922974
commit 4e8ae49f60
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/build.yml
+3 -3
View File
@@ -31,7 +31,7 @@ jobs:
name: Build wheels and Upload name: Build wheels and Upload
needs: create_releases needs: create_releases
runs-on: ubuntu-22.04 runs-on: ubuntu-22.04
timeout-minutes: 6000 timeout-minutes: 700
strategy: strategy:
fail-fast: false fail-fast: false
matrix: matrix:
@@ -137,12 +137,12 @@ jobs:
${{ env.CACHE_KEY }}- ${{ env.CACHE_KEY }}-
- name: Build wheels - name: Build wheels
timeout-minutes: 6000 timeout-minutes: 600
run: | run: |
pip install setuptools==68.0.0 ninja packaging wheel pip install setuptools==68.0.0 ninja packaging wheel
export PATH=/usr/local/nvidia/bin:/usr/local/nvidia/lib64:$PATH export PATH=/usr/local/nvidia/bin:/usr/local/nvidia/lib64:$PATH
export LD_LIBRARY_PATH=/usr/local/nvidia/lib64:/usr/local/cuda/lib64:$LD_LIBRARY_PATH export LD_LIBRARY_PATH=/usr/local/nvidia/lib64:/usr/local/cuda/lib64:$LD_LIBRARY_PATH
export MAX_JOBS=$(($(nproc) - 1)) export MAX_JOBS=$(nproc)
cd flash-attention cd flash-attention
FLASH_ATTENTION_FORCE_BUILD="TRUE" python setup.py bdist_wheel --dist-dir=dist FLASH_ATTENTION_FORCE_BUILD="TRUE" python setup.py bdist_wheel --dist-dir=dist
base_wheel_name=$(basename $(ls dist/*.whl | head -n 1)) base_wheel_name=$(basename $(ls dist/*.whl | head -n 1))