Skip to content

Commit b2f5d53

Browse files
vuksan314jeffdaily
authored andcommitted
[release/2.9] Add gfx110X and gfx115X to prefered hipBLASLt list (#2742)
hipblaslt should provide better performance in general
1 parent e7d49c3 commit b2f5d53

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
@@ -457,8 +457,11 @@ at::BlasBackend Context::blasPreferredBackend() {
457457
static const bool hipblaslt_preferred = []() {
458458
static const std::vector<std::string> archs = {
459459
"gfx90a", "gfx942",
460-
#if ROCM_VERSION >= 60400
461-
"gfx1200", "gfx1201",
460+
#if ROCM_VERSION >= 60300
461+
"gfx1100", "gfx1101", "gfx1102", "gfx1200", "gfx1201",
462+
#endif
463+
#if ROCM_VERSION >= 60402
464+
"gfx1150", "gfx1151",
462465
#endif
463466
#if ROCM_VERSION >= 60500
464467
"gfx950"
@@ -488,7 +491,10 @@ at::BlasBackend Context::blasPreferredBackend() {
488491
static const std::vector<std::string> archs = {
489492
"gfx90a", "gfx942",
490493
#if ROCM_VERSION >= 60300
491-
"gfx1100", "gfx1101", "gfx1200", "gfx1201", "gfx908",
494+
"gfx1100", "gfx1101", "gfx1102", "gfx1200", "gfx1201", "gfx908",
495+
#endif
496+
#if ROCM_VERSION >= 60402
497+
"gfx1150", "gfx1151",
492498
#endif
493499
#if ROCM_VERSION >= 60500
494500
"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", "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)