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:27:54 -04:00
Code Issues Packages Projects Releases 2 Wiki Activity

feat: Added time command

Browse Source
This commit is contained in:
Junya Morioka
2025-12-04 19:16:37 +09:00
parent 0bce0110f7
commit 4db0ed0589
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
build_linux.sh
+3 -2
View File
@@ -74,7 +74,8 @@ echo " NVCC_THREADS: $NVCC_THREADS"
echo "Building wheels..." echo "Building wheels..."
cd flash-attention cd flash-attention
LOCAL_VERSION_LABEL="cu${MATRIX_CUDA_VERSION}torch${MATRIX_TORCH_VERSION}" 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_THREADS=$NVCC_THREADS MAX_JOBS=$MAX_JOBS \
python setup.py bdist_wheel --dist-dir=dist FLASH_ATTENTION_FORCE_BUILD=TRUE FLASH_ATTN_LOCAL_VERSION=${LOCAL_VERSION_LABEL} \
time python setup.py bdist_wheel --dist-dir=dist
wheel_name=$(basename $(ls dist/*.whl | head -n 1)) wheel_name=$(basename $(ls dist/*.whl | head -n 1))
echo "Built wheel: $wheel_name" echo "Built wheel: $wheel_name"