From 5021b1371e40505cf2044a50d9da98c29a3cb842 Mon Sep 17 00:00:00 2001 From: Milica Stankovic Date: Tue, 26 Aug 2025 17:32:04 +0200 Subject: [PATCH 1/3] Add gfx1150/gfx1151 to hipBLASLt supported architecture list --- aten/src/ATen/Context.cpp | 2 +- aten/src/ATen/native/cuda/Blas.cpp | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/aten/src/ATen/Context.cpp b/aten/src/ATen/Context.cpp index fefcf731cd82..a1b3e998fa06 100644 --- a/aten/src/ATen/Context.cpp +++ b/aten/src/ATen/Context.cpp @@ -362,7 +362,7 @@ at::BlasBackend Context::blasPreferredBackend() { static const std::vector archs = { "gfx90a", "gfx942", #if ROCM_VERSION >= 60300 - "gfx1100", "gfx1101", "gfx1200", "gfx1201", + "gfx1100", "gfx1101", "gfx1150", "gfx1151", "gfx1200", "gfx1201", #endif #if ROCM_VERSION >= 60500 "gfx950" diff --git a/aten/src/ATen/native/cuda/Blas.cpp b/aten/src/ATen/native/cuda/Blas.cpp index 1fc9e14189e4..ee6f7cf53b03 100644 --- a/aten/src/ATen/native/cuda/Blas.cpp +++ b/aten/src/ATen/native/cuda/Blas.cpp @@ -270,7 +270,7 @@ static bool isSupportedHipLtROCmArch(int index) { static const std::vector archs = { "gfx90a", "gfx942", #if ROCM_VERSION >= 60300 - "gfx1100", "gfx1101", "gfx1200", "gfx1201", + "gfx1100", "gfx1101", "gfx1150", "gfx1151", "gfx1200", "gfx1201", #endif #if ROCM_VERSION >= 60500 "gfx950" @@ -1049,7 +1049,7 @@ static bool _scaled_mm_allowed_device(bool sm90_only=false) { static const std::vector archs = { "gfx942", #if ROCM_VERSION >= 60300 - "gfx1200", "gfx1201", + "gfx1150", "gfz1151", "gfx1200", "gfx1201", #endif #if ROCM_VERSION >= 60500 "gfx950" From d599376f3604a046af0130018c6e77f04b12b641 Mon Sep 17 00:00:00 2001 From: Milica Stankovic Date: Tue, 26 Aug 2025 18:30:15 +0200 Subject: [PATCH 2/3] Small fix --- aten/src/ATen/Context.cpp | 4 ++-- aten/src/ATen/native/cuda/Blas.cpp | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/aten/src/ATen/Context.cpp b/aten/src/ATen/Context.cpp index a1b3e998fa06..68ed4a8a054f 100644 --- a/aten/src/ATen/Context.cpp +++ b/aten/src/ATen/Context.cpp @@ -336,7 +336,7 @@ at::BlasBackend Context::blasPreferredBackend() { static const std::vector archs = { "gfx90a", "gfx942", #if ROCM_VERSION >= 60400 - "gfx1200", "gfx1201", + "gfx1150"{, "gfx1151", }"gfx1200", "gfx1201", #endif #if ROCM_VERSION >= 60500 "gfx950" @@ -362,7 +362,7 @@ at::BlasBackend Context::blasPreferredBackend() { static const std::vector archs = { "gfx90a", "gfx942", #if ROCM_VERSION >= 60300 - "gfx1100", "gfx1101", "gfx1150", "gfx1151", "gfx1200", "gfx1201", + "gfx1100", "gfx1101", "gfx1200", "gfx1201", #endif #if ROCM_VERSION >= 60500 "gfx950" diff --git a/aten/src/ATen/native/cuda/Blas.cpp b/aten/src/ATen/native/cuda/Blas.cpp index ee6f7cf53b03..2fc9cdddaf66 100644 --- a/aten/src/ATen/native/cuda/Blas.cpp +++ b/aten/src/ATen/native/cuda/Blas.cpp @@ -1049,7 +1049,7 @@ static bool _scaled_mm_allowed_device(bool sm90_only=false) { static const std::vector archs = { "gfx942", #if ROCM_VERSION >= 60300 - "gfx1150", "gfz1151", "gfx1200", "gfx1201", + "gfx1200", "gfx1201", #endif #if ROCM_VERSION >= 60500 "gfx950" From c54aa1530f664cf28e9675733f869806d79ec45f Mon Sep 17 00:00:00 2001 From: Milica Stankovic Date: Thu, 28 Aug 2025 11:21:15 +0200 Subject: [PATCH 3/3] Remove { character --- aten/src/ATen/Context.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aten/src/ATen/Context.cpp b/aten/src/ATen/Context.cpp index 68ed4a8a054f..1a679078281e 100644 --- a/aten/src/ATen/Context.cpp +++ b/aten/src/ATen/Context.cpp @@ -336,7 +336,7 @@ at::BlasBackend Context::blasPreferredBackend() { static const std::vector archs = { "gfx90a", "gfx942", #if ROCM_VERSION >= 60400 - "gfx1150"{, "gfx1151", }"gfx1200", "gfx1201", + "gfx1150", "gfx1151", "gfx1200", "gfx1201", #endif #if ROCM_VERSION >= 60500 "gfx950"