Skip to content

Commit 63e525b

Browse files
authored
[release/2.8] Add gfx110X and gfx115X to prefered hipBLASLt list (#2741)
Switching to hipblaslt in general improve interface performance.
1 parent 5729657 commit 63e525b

File tree

2 files changed

+12
-3
lines changed

2 files changed

+12
-3
lines changed

aten/src/ATen/Context.cpp

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -336,7 +336,10 @@ 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", "gfx1102", "gfx1200", "gfx1201",
340+
#endif
341+
#if ROCM_VERSION >= 60402
342+
"gfx1150", "gfx1151",
340343
#endif
341344
#if ROCM_VERSION >= 60402
342345
"gfx1150", "gfx1151",
@@ -365,7 +368,10 @@ at::BlasBackend Context::blasPreferredBackend() {
365368
static const std::vector<std::string> archs = {
366369
"gfx90a", "gfx942",
367370
#if ROCM_VERSION >= 60300
368-
"gfx1100", "gfx1101", "gfx1200", "gfx1201",
371+
"gfx1100", "gfx1101", "gfx1102", "gfx1200", "gfx1201",
372+
#endif
373+
#if ROCM_VERSION >= 60402
374+
"gfx1150", "gfx1151",
369375
#endif
370376
#if ROCM_VERSION >= 60402
371377
"gfx1150", "gfx1151",

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

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -270,7 +270,10 @@ 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", "gfx1102", "gfx1200", "gfx1201",
274+
#endif
275+
#if ROCM_VERSION >= 60402
276+
"gfx1150", "gfx1151",
274277
#endif
275278
#if ROCM_VERSION >= 60402
276279
"gfx1150", "gfx1151",

0 commit comments

Comments
 (0)