Skip to content

Commit f1e6ec9

Browse files
committed
add gfx1152 and gfx1102
1 parent 8b8ee7f commit f1e6ec9

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-
"gfx1100", "gfx1101", "gfx1151", "gfx1200", "gfx1201",
339+
"gfx1100", "gfx1101", "gfx1102", "gfx1200", "gfx1201",
340+
#endif
341+
#if ROCM_VERSION >= 60402
342+
"gfx1150", "gfx1151",
340343
#endif
341344
#if ROCM_VERSION >= 60500
342345
"gfx950"
@@ -362,7 +365,10 @@ at::BlasBackend Context::blasPreferredBackend() {
362365
static const std::vector<std::string> archs = {
363366
"gfx90a", "gfx942",
364367
#if ROCM_VERSION >= 60300
365-
"gfx1100", "gfx1101", "gfx1151", "gfx1200", "gfx1201",
368+
"gfx1100", "gfx1101", "gfx1102", "gfx1200", "gfx1201",
369+
#endif
370+
#if ROCM_VERSION >= 60402
371+
"gfx1150", "gfx1151",
366372
#endif
367373
#if ROCM_VERSION >= 60500
368374
"gfx950"

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", "gfx1151", "gfx1200", "gfx1201",
273+
"gfx1100", "gfx1101", "gfx1102", "gfx1200", "gfx1201",
274+
#endif
275+
#if ROCM_VERSION >= 60402
276+
"gfx1150", "gfx1151",
274277
#endif
275278
#if ROCM_VERSION >= 60500
276279
"gfx950"

0 commit comments

Comments
 (0)