yindo/flash-attention
1
0
Fork 0
mirror of https://github.com/BillyOutlast/flash-attention.git synced 2026-06-30 21:07:55 -04:00
Code Issues Packages Projects Releases Wiki Activity

[AMD ROCm] Support MI350 (#1586)

Browse Source 
* enable gfx950 support

* update ck for gfx950

---------

Co-authored-by: illsilin <Illia.Silin@amd.com>
This commit is contained in:
rocking
2025-04-12 02:06:03 +08:00
committed by GitHub
parent 7ff73af43f
commit c1352b6d96
3 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.gitmodules
+1
View File
@@ -4,3 +4,4 @@
[submodule "csrc/composable_kernel"]
path = csrc/composable_kernel
url = https://github.com/ROCm/composable_kernel.git
branch = amd-master
csrc/composable_kernel
+1 -1
Submodule csrc/composable_kernel updated: 888317e698...72c0261ef1
setup.py
+1 -1
View File
@@ -132,7 +132,7 @@ def rename_cpp_to_cu(cpp_files):
def validate_and_update_archs(archs):
# List of allowed architectures
allowed_archs = ["native", "gfx90a", "gfx940", "gfx941", "gfx942"]
allowed_archs = ["native", "gfx90a", "gfx950", "gfx942"]
# Validate if each element in archs is in allowed_archs
assert all(