Skip to content

Commit d631bdb

Browse files
[release/2.8] Add gfx1150/gfx1151 to hipBLASLt supported architecture list (#2582)
Co-authored-by: Slobodan Josic <[email protected]>
1 parent 71347e3 commit d631bdb

File tree

2 files changed

+9
-0
lines changed

2 files changed

+9
-0
lines changed

aten/src/ATen/Context.cpp

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -338,6 +338,9 @@ at::BlasBackend Context::blasPreferredBackend() {
338338
#if ROCM_VERSION >= 60400
339339
"gfx1200", "gfx1201",
340340
#endif
341+
#if ROCM_VERSION >= 60402
342+
"gfx1150", "gfx1151",
343+
#endif
341344
#if ROCM_VERSION >= 60500
342345
"gfx950"
343346
#endif
@@ -364,6 +367,9 @@ at::BlasBackend Context::blasPreferredBackend() {
364367
#if ROCM_VERSION >= 60300
365368
"gfx1100", "gfx1101", "gfx1200", "gfx1201",
366369
#endif
370+
#if ROCM_VERSION >= 60402
371+
"gfx1150", "gfx1151",
372+
#endif
367373
#if ROCM_VERSION >= 60500
368374
"gfx950"
369375
#endif

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

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -272,6 +272,9 @@ static bool isSupportedHipLtROCmArch(int index) {
272272
#if ROCM_VERSION >= 60300
273273
"gfx1100", "gfx1101", "gfx1200", "gfx1201",
274274
#endif
275+
#if ROCM_VERSION >= 60402
276+
"gfx1150", "gfx1151",
277+
#endif
275278
#if ROCM_VERSION >= 60500
276279
"gfx950"
277280
#endif

0 commit comments

Comments
 (0)