mirror of
https://github.com/mudler/LocalAI.git
synced 2026-09-12 22:33:54 -04:00
fix(qwen-tts): limit CUDA compiler threads
The CUDA 13 FlashAttention build still exhausts hosted-runner memory with a single ninja worker because nvcc can compile multiple threads internally. Limit nvcc to one thread for that profile and guard the setting in the backend test script. Assisted-by: Codex:gpt-5 Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
This commit is contained in:
1 parent
43c54bc9ba
commit
1dfa25e681
3 files changed
+18
No files matched your search
@@ -10,4 +10,11 @@ else
|
||||
source $backend_dir/../common/libbackend.sh
|
||||
fi
|
||||
|
||||
# CUDA 13 has no prebuilt FlashAttention wheel, so the fallback source build
|
||||
# exceeds the CI runner's memory when ninja compiles multiple units at once.
|
||||
if [ "x${BUILD_PROFILE}" = "xcublas13" ]; then
|
||||
export MAX_JOBS="${MAX_JOBS:-1}"
|
||||
export NVCC_THREADS="${NVCC_THREADS:-1}"
|
||||
fi
|
||||
|
||||
installRequirements
|
||||
@@ -0,0 +1,2 @@
|
||||
ninja
|
||||
flash-attn
|
||||
@@ -2,6 +2,15 @@
|
||||
set -e
|
||||
|
||||
backend_dir=$(dirname $0)
|
||||
|
||||
for cuda_version in 12 13; do
|
||||
grep -qx "flash-attn" "$backend_dir/requirements-cublas${cuda_version}-after.txt"
|
||||
done
|
||||
|
||||
grep -q 'BUILD_PROFILE.*cublas13' "$backend_dir/install.sh"
|
||||
grep -q 'MAX_JOBS.*1' "$backend_dir/install.sh"
|
||||
grep -q 'NVCC_THREADS.*1' "$backend_dir/install.sh"
|
||||
|
||||
if [ -d $backend_dir/common ]; then
|
||||
source $backend_dir/common/libbackend.sh
|
||||
else
|
||||
|
||||
Reference in new issue
Block a user