From 44652b848b5c4bb03b3b4f9c6bafd94dfc21e286 Mon Sep 17 00:00:00 2001 From: nemanjaudovic Date: Wed, 17 Jun 2026 12:00:31 -0400 Subject: [PATCH] Add gfx1100, gfx1101, gfx1151 to hipBLASLt preferred archs --- aten/src/ATen/cuda/detail/CUDAHooks.cpp | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/aten/src/ATen/cuda/detail/CUDAHooks.cpp b/aten/src/ATen/cuda/detail/CUDAHooks.cpp index 5f81407b1ac03..f6da4bcf78232 100644 --- a/aten/src/ATen/cuda/detail/CUDAHooks.cpp +++ b/aten/src/ATen/cuda/detail/CUDAHooks.cpp @@ -554,7 +554,10 @@ const std::vector& CUDAHooks::getHipblasltPreferredArchs() const { "gfx1200", "gfx1201", #endif #if ROCM_VERSION >= 70000 - "gfx950" + "gfx950", +#endif +#if ROCM_VERSION >= 71300 + "gfx1100", "gfx1101", "gfx1151" #endif }; return archs;