Skip to content

Commit 5021b13

Browse files
committed
Add gfx1150/gfx1151 to hipBLASLt supported architecture list
1 parent 0def0b8 commit 5021b13

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

aten/src/ATen/Context.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -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", "gfx1150", "gfx1151", "gfx1200", "gfx1201",
366366
#endif
367367
#if ROCM_VERSION >= 60500
368368
"gfx950"

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

Lines changed: 2 additions & 2 deletions
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"
@@ -1049,7 +1049,7 @@ static bool _scaled_mm_allowed_device(bool sm90_only=false) {
10491049
static const std::vector<std::string> archs = {
10501050
"gfx942",
10511051
#if ROCM_VERSION >= 60300
1052-
"gfx1200", "gfx1201",
1052+
"gfx1150", "gfz1151", "gfx1200", "gfx1201",
10531053
#endif
10541054
#if ROCM_VERSION >= 60500
10551055
"gfx950"

0 commit comments

Comments
 (0)