From 2784db02fe62ce687ad818dca7b754c97e1a5ad7 Mon Sep 17 00:00:00 2001 From: illsilin Date: Mon, 31 Mar 2025 14:16:35 -0700 Subject: [PATCH 1/2] enable gfx950 support --- .gitmodules | 1 + setup.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/.gitmodules b/.gitmodules index 6216182e721..a6446cc597a 100644 --- a/.gitmodules +++ b/.gitmodules @@ -4,3 +4,4 @@ [submodule "csrc/composable_kernel"] path = csrc/composable_kernel url = https://github.com/ROCm/composable_kernel.git + branch = amd-master diff --git a/setup.py b/setup.py index a802a7e65e4..fdef39deb48 100644 --- a/setup.py +++ b/setup.py @@ -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( From fd56098d049940d0bae3eacaabb2663566f0c702 Mon Sep 17 00:00:00 2001 From: rocking Date: Tue, 8 Apr 2025 08:29:26 +0000 Subject: [PATCH 2/2] update ck for gfx950 --- csrc/composable_kernel | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/csrc/composable_kernel b/csrc/composable_kernel index 888317e698e..72c0261ef1b 160000 --- a/csrc/composable_kernel +++ b/csrc/composable_kernel @@ -1 +1 @@ -Subproject commit 888317e698e9803c62bd38568abc9e05d7709f33 +Subproject commit 72c0261ef1b40587ee8674b9d49b4fd6b46b0335