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: Update build parallel

Browse Source
This commit is contained in:
Junya Morioka
2025-12-05 12:33:24 +09:00
parent 3f85abe6ad
commit 20a9205e2c
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/test-utils.yml
+2 -2
View File
@@ -51,9 +51,9 @@ jobs:
BASE_THREADS=$(awk -v max="$MAX_PRODUCT" 'BEGIN {print int(sqrt(max))}') BASE_THREADS=$(awk -v max="$MAX_PRODUCT" 'BEGIN {print int(sqrt(max))}')
if awk "BEGIN {exit !($RAM_GB <= 16)}"; then if awk "BEGIN {exit !($RAM_GB <= 16)}"; then
# If RAM is less than 16GB, set NVCC_THREADS to 1 # If RAM is less than 16GB, set NVCC_THREADS to 1 and MAX_JOBS to 2
NVCC_THREADS=1 NVCC_THREADS=1
MAX_JOBS=$((MAX_PRODUCT / NVCC_THREADS)) MAX_JOBS=2
elif (( BASE_THREADS <= 4 )); then elif (( BASE_THREADS <= 4 )); then
NVCC_THREADS=$BASE_THREADS NVCC_THREADS=$BASE_THREADS
MAX_JOBS=$BASE_THREADS MAX_JOBS=$BASE_THREADS
build_linux.sh
+2 -2
View File
@@ -66,9 +66,9 @@ if [[ -z "${MAX_JOBS:-}" && -z "${NVCC_THREADS:-}" ]]; then
BASE_THREADS=$(awk -v max="$MAX_PRODUCT" 'BEGIN {print int(sqrt(max))}') BASE_THREADS=$(awk -v max="$MAX_PRODUCT" 'BEGIN {print int(sqrt(max))}')
if awk "BEGIN {exit !($RAM_GB <= 16)}"; then if awk "BEGIN {exit !($RAM_GB <= 16)}"; then
# If RAM is less than 16GB, set NVCC_THREADS to 1 # If RAM is less than 16GB, set NVCC_THREADS to 1 and MAX_JOBS to 2
NVCC_THREADS=1 NVCC_THREADS=1
MAX_JOBS=$((MAX_PRODUCT / NVCC_THREADS)) MAX_JOBS=2
elif (( BASE_THREADS <= 4 )); then elif (( BASE_THREADS <= 4 )); then
NVCC_THREADS=$BASE_THREADS NVCC_THREADS=$BASE_THREADS
MAX_JOBS=$BASE_THREADS MAX_JOBS=$BASE_THREADS