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: remove rename for local version

Browse Source
This commit is contained in:
Junya Morioka
2025-11-26 19:43:55 +09:00
parent ba2db41266
commit 28f8b7071d
2 changed files with 5 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
build_linux.sh
+1 -3
View File
@@ -49,7 +49,5 @@ 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}"
FLASH_ATTENTION_FORCE_BUILD=TRUE FLASH_ATTN_LOCAL_VERSION=${LOCAL_VERSION_LABEL} python setup.py bdist_wheel --dist-dir=dist FLASH_ATTENTION_FORCE_BUILD=TRUE FLASH_ATTN_LOCAL_VERSION=${LOCAL_VERSION_LABEL} python setup.py bdist_wheel --dist-dir=dist
base_wheel_name=$(basename $(ls dist/*.whl | head -n 1)) wheel_name=$(basename $(ls dist/*.whl | head -n 1))
wheel_name=$(echo $base_wheel_name | sed "s/$FLASH_ATTN_VERSION/$FLASH_ATTN_VERSION+cu${MATRIX_CUDA_VERSION}torch${MATRIX_TORCH_VERSION}/")
mv -v dist/$base_wheel_name dist/$wheel_name
echo "Built wheel: $wheel_name" echo "Built wheel: $wheel_name"
build_windows.ps1
+4 -7
View File
@@ -1,13 +1,13 @@
param( param(
[Parameter(Mandatory=$true)] [Parameter(Mandatory=$true)]
[string]$FlashAttnVersion, [string]$FlashAttnVersion,
[Parameter(Mandatory=$true)] [Parameter(Mandatory=$true)]
[string]$PythonVersion, [string]$PythonVersion,
[Parameter(Mandatory=$true)] [Parameter(Mandatory=$true)]
[string]$TorchVersion, [string]$TorchVersion,
[Parameter(Mandatory=$true)] [Parameter(Mandatory=$true)]
[string]$CudaVersion [string]$CudaVersion
) )
@@ -72,8 +72,5 @@ $env:FLASH_ATTN_LOCAL_VERSION = "cu$MatrixCudaVersion" + "torch$MatrixTorchVersi
cd flash-attention cd flash-attention
python setup.py bdist_wheel --dist-dir=dist python setup.py bdist_wheel --dist-dir=dist
$baseWheelName = Get-ChildItem -Path "dist\*.whl" | Select-Object -First 1 | ForEach-Object { $_.Name } $WheelName = Get-ChildItem -Path "dist\*.whl" | Select-Object -First 1 | ForEach-Object { $_.Name }
$wheelName = $baseWheelName.Replace($FlashAttnVersion, "$FlashAttnVersion+cu$MatrixCudaVersion" + "torch$MatrixTorchVersion")
Move-Item "dist\$baseWheelName" "dist\$wheelName"
Write-Host "Built wheel: $wheelName" Write-Host "Built wheel: $wheelName"
cd ..