Skip to content

Commit b8bf963

Browse files
committed
Add gfx1150/gfx1151 to hipBLASLt supported architecture list
1 parent 7b2a4fd commit b8bf963

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

aten/src/ATen/Context.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -336,7 +336,7 @@ at::BlasBackend Context::blasPreferredBackend() {
336336
static const std::vector<std::string> archs = {
337337
"gfx90a", "gfx942",
338338
#if ROCM_VERSION >= 60400
339-
"gfx1200", "gfx1201",
339+
"gfx1150", "gfx1151", "gfx1200", "gfx1201",
340340
#endif
341341
#if ROCM_VERSION >= 60500
342342
"gfx950"

aten/src/ATen/native/cuda/Blas.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -270,7 +270,7 @@ static bool isSupportedHipLtROCmArch(int index) {
270270
static const std::vector<std::string> archs = {
271271
"gfx90a", "gfx942",
272272
#if ROCM_VERSION >= 60300
273-
"gfx1100", "gfx1101", "gfx1200", "gfx1201",
273+
"gfx1100", "gfx1101", "gfx1150", "gfx1151", "gfx1200", "gfx1201",
274274
#endif
275275
#if ROCM_VERSION >= 60500
276276
"gfx950"

0 commit comments

Comments
 (0)