Skip to content

Commit 3cdddfe

Browse files
authored
[release/2.7] Prefer hipblaslt for gfx1200, gfx1201 (#2125)
1 parent 9009197 commit 3cdddfe

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

aten/src/ATen/Context.cpp

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -335,6 +335,9 @@ at::BlasBackend Context::blasPreferredBackend() {
335335
static const bool hipblaslt_preferred = []() {
336336
static const std::vector<std::string> archs = {
337337
"gfx90a", "gfx942",
338+
#if ROCM_VERSION >= 60300
339+
"gfx1200", "gfx1201",
340+
#endif
338341
#if ROCM_VERSION >= 60500
339342
"gfx950"
340343
#endif

0 commit comments

Comments
 (0)