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: add --allow-unsupported-compiler flag for nvcc

Browse Source 
- Add NVCC_APPEND_FLAGS="--allow-unsupported-compiler" to Linux build
  script to support gcc-toolset-14 in manylinux_2_28 containers
- Resolves CUDA 12.6 compilation failure with gcc versions later than 13
This commit is contained in:
Junya Morioka
2026-01-28 01:18:29 +09:00
parent d01b5aee34
commit 78042bb8c2
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
build_linux.sh
+1
View File
@@ -92,6 +92,7 @@ cd flash-attention
LOCAL_VERSION_LABEL="cu${MATRIX_CUDA_VERSION}torch${MATRIX_TORCH_VERSION}"
NVCC_THREADS=$NVCC_THREADS MAX_JOBS=$MAX_JOBS \
FLASH_ATTENTION_FORCE_BUILD=TRUE FLASH_ATTN_LOCAL_VERSION=${LOCAL_VERSION_LABEL} \
NVCC_APPEND_FLAGS="--allow-unsupported-compiler" \
time python setup.py bdist_wheel --dist-dir=dist
wheel_name=$(basename $(ls dist/*.whl | head -n 1))
echo "Built wheel: $wheel_name"