From af33880defa4615ef5ed71f1fbe752d2c7482608 Mon Sep 17 00:00:00 2001 From: nemanjaudovic Date: Wed, 17 Jun 2026 12:06:03 -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 03a3a97525a4..96af4e98e679 100644 --- a/aten/src/ATen/cuda/detail/CUDAHooks.cpp +++ b/aten/src/ATen/cuda/detail/CUDAHooks.cpp @@ -558,7 +558,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;