yindo/flash-attention-prebuild-wheels-rocm
1
0
Fork 0
mirror of https://github.com/BillyOutlast/flash-attention-prebuild-wheels-rocm.git synced 2026-06-30 23:57:53 -04:00
Code Issues Packages Projects Releases 2 Wiki Activity

[WIP] update linux workflow

Browse Source
This commit is contained in:
Junya Morioka
2025-06-08 06:30:50 +09:00
parent b5ca5ec9ff
commit bd41ed80a2
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
build_linux.sh
+4 -2
View File
@@ -36,8 +36,10 @@ TORCH_CUDA_VERSION=$(python -c "from os import environ as env; \
'2.7': [118, 126, 128], \
'2.8': [128], \
}; \
target_cuda_versions = support_cuda_versions[env['MATRIX_TORCH_VERSION']]; \
cuda_version = int(env['MATRIX_CUDA_VERSION']); \
matrix_cuda_version = '$MATRIX_CUDA_VERSION'; \
matrix_torch_version = '$MATRIX_TORCH_VERSION'; \
target_cuda_versions = support_cuda_versions[matrix_torch_version]; \
cuda_version = int(matrix_cuda_version); \
closest_version = min(target_cuda_versions, key=lambda x: abs(x - cuda_version)); \
print(closest_version) \
")