Skip to content

Commit 0a0b41b

Browse files
committed
updated supported and hipblaslt preffered architures
1 parent 7a52036 commit 0a0b41b

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

aten/src/ATen/Context.cpp

Lines changed: 2 additions & 2 deletions
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+
"gfx1100", "gfx1101", "gfx1151", "gfx1200", "gfx1201",
340340
#endif
341341
#if ROCM_VERSION >= 60500
342342
"gfx950"
@@ -362,7 +362,7 @@ at::BlasBackend Context::blasPreferredBackend() {
362362
static const std::vector<std::string> archs = {
363363
"gfx90a", "gfx942",
364364
#if ROCM_VERSION >= 60300
365-
"gfx1100", "gfx1101", "gfx1200", "gfx1201",
365+
"gfx1100", "gfx1101", "gfx1151", "gfx1200", "gfx1201",
366366
#endif
367367
#if ROCM_VERSION >= 60500
368368
"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", "gfx1151", "gfx1200", "gfx1201",
274274
#endif
275275
#if ROCM_VERSION >= 60500
276276
"gfx950"

0 commit comments

Comments
 (0)