From 4db0ed0589cb29f7e6ee6522f7f678b41b062e55 Mon Sep 17 00:00:00 2001 From: Junya Morioka Date: Thu, 4 Dec 2025 19:16:37 +0900 Subject: [PATCH] feat: Added time command --- build_linux.sh | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/build_linux.sh b/build_linux.sh index caa9f03..b528bd4 100755 --- a/build_linux.sh +++ b/build_linux.sh @@ -74,7 +74,8 @@ echo " NVCC_THREADS: $NVCC_THREADS" echo "Building wheels..." 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} \ - python setup.py bdist_wheel --dist-dir=dist +NVCC_THREADS=$NVCC_THREADS MAX_JOBS=$MAX_JOBS \ + 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)) echo "Built wheel: $wheel_name"