Skip to content

Commit 96d0844

Browse files
committed
update support/prefer hipblaslt architectures
1 parent caf1b51 commit 96d0844

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
@@ -472,7 +472,7 @@ at::BlasBackend Context::blasPreferredBackend() {
472472
static const std::vector<std::string> archs = {
473473
"gfx90a", "gfx942",
474474
#if ROCM_VERSION >= 60400
475-
"gfx1200", "gfx1201",
475+
"gfx1100", "gfx1101", "gfx1151", "gfx1200", "gfx1201",
476476
#endif
477477
#if ROCM_VERSION >= 60500
478478
"gfx950"
@@ -502,7 +502,7 @@ at::BlasBackend Context::blasPreferredBackend() {
502502
static const std::vector<std::string> archs = {
503503
"gfx90a", "gfx942",
504504
#if ROCM_VERSION >= 60300
505-
"gfx1100", "gfx1101", "gfx1200", "gfx1201", "gfx908",
505+
"gfx1100", "gfx1101", "gfx1151", "gfx1200", "gfx1201", "gfx908",
506506
#endif
507507
#if ROCM_VERSION >= 60500
508508
"gfx950"

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -283,7 +283,7 @@ static bool isSupportedHipLtROCmArch(int index) {
283283
static const std::vector<std::string> archs = {
284284
"gfx90a", "gfx942",
285285
#if ROCM_VERSION >= 60300
286-
"gfx1100", "gfx1101", "gfx1200", "gfx1201", "gfx908",
286+
"gfx1100", "gfx1101", "gfx1151", "gfx1200", "gfx1201", "gfx908",
287287
#endif
288288
#if ROCM_VERSION >= 60500
289289
"gfx950"

0 commit comments

Comments
 (0)