Skip to content

Commit 151b746

Browse files
committed
add gfx1151 and gfx1102
1 parent 96d0844 commit 151b746

File tree

2 files changed

+13
-4
lines changed

2 files changed

+13
-4
lines changed

aten/src/ATen/Context.cpp

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -471,8 +471,11 @@ at::BlasBackend Context::blasPreferredBackend() {
471471
static const bool hipblaslt_preferred = []() {
472472
static const std::vector<std::string> archs = {
473473
"gfx90a", "gfx942",
474-
#if ROCM_VERSION >= 60400
475-
"gfx1100", "gfx1101", "gfx1151", "gfx1200", "gfx1201",
474+
#if ROCM_VERSION >= 60300
475+
"gfx1100", "gfx1101", "gfx1102", "gfx1200", "gfx1201",
476+
#endif
477+
#if ROCM_VERSION >= 60402
478+
"gfx1150", "gfx1151",
476479
#endif
477480
#if ROCM_VERSION >= 60500
478481
"gfx950"
@@ -502,7 +505,10 @@ at::BlasBackend Context::blasPreferredBackend() {
502505
static const std::vector<std::string> archs = {
503506
"gfx90a", "gfx942",
504507
#if ROCM_VERSION >= 60300
505-
"gfx1100", "gfx1101", "gfx1151", "gfx1200", "gfx1201", "gfx908",
508+
"gfx1100", "gfx1101", "gfx1102", "gfx1200", "gfx1201", "gfx908",
509+
#endif
510+
#if ROCM_VERSION >= 60402
511+
"gfx1150", "gfx1151",
506512
#endif
507513
#if ROCM_VERSION >= 60500
508514
"gfx950"

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

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -283,7 +283,10 @@ static bool isSupportedHipLtROCmArch(int index) {
283283
static const std::vector<std::string> archs = {
284284
"gfx90a", "gfx942",
285285
#if ROCM_VERSION >= 60300
286-
"gfx1100", "gfx1101", "gfx1151", "gfx1200", "gfx1201", "gfx908",
286+
"gfx1100", "gfx1101", "gfx1102", "gfx1200", "gfx1201", "gfx908",
287+
#endif
288+
#if ROCM_VERSION >= 60402
289+
"gfx1150", "gfx1151",
287290
#endif
288291
#if ROCM_VERSION >= 60500
289292
"gfx950"

0 commit comments

Comments
 (0)