From 4e8ae49f60dac949efcf5440e441d2874f480920 Mon Sep 17 00:00:00 2001 From: Junya Morioka Date: Sun, 27 Apr 2025 23:00:33 +0900 Subject: [PATCH] fix set timeout-minutes, update MAX_JOBS --- .github/workflows/build.yml | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index 25fe43c..0d31ea9 100644 --- a/.github/workflows/build.yml +++ b/.github/workflows/build.yml @@ -31,7 +31,7 @@ jobs: name: Build wheels and Upload needs: create_releases runs-on: ubuntu-22.04 - timeout-minutes: 6000 + timeout-minutes: 700 strategy: fail-fast: false matrix: @@ -137,12 +137,12 @@ jobs: ${{ env.CACHE_KEY }}- - name: Build wheels - timeout-minutes: 6000 + timeout-minutes: 600 run: | pip install setuptools==68.0.0 ninja packaging wheel 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 MAX_JOBS=$(($(nproc) - 1)) + export MAX_JOBS=$(nproc) cd flash-attention FLASH_ATTENTION_FORCE_BUILD="TRUE" python setup.py bdist_wheel --dist-dir=dist base_wheel_name=$(basename $(ls dist/*.whl | head -n 1))