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/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
diff --git a/setup.py b/setup.py
index 264b0eed511..2430f4c6d5d 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(