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

add torch==2.9.0

Browse Source
This commit is contained in:
Junya Morioka
2025-09-20 11:15:05 +09:00
parent 26d894259f
commit bbad4182f4
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
get_torch_cuda_version.py
+1
View File
@@ -10,6 +10,7 @@ support_cuda_versions = {
"2.6": [118, 124, 126],
"2.7": [118, 126, 128],
"2.8": [126, 128, 129],
"2.9": [126, 128, 129],
}
cuda_version = int(sys.argv[1])