Skip to content

Commit 9030e36

Browse files
apakbinpruthvistony
authored andcommitted
Change gfx110x BLAS preferred backend
1 parent 1cb818b commit 9030e36

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -257,9 +257,9 @@ static bool isSupportedHipLtROCmArch(int index) {
257257
hipDeviceProp_t* prop = at::cuda::getDeviceProperties(index);
258258
std::string device_arch = prop->gcnArchName;
259259
static const std::vector<std::string> archs = {
260-
"gfx90a", "gfx942",
260+
"gfx90a", "gfx942"
261261
#if ROCM_VERSION >= 60300
262-
"gfx1100", "gfx1101", "gfx1200", "gfx1201"
262+
, "gfx1200", "gfx1201"
263263
#endif
264264
#if ROCM_VERSION >= 60500
265265
"gfx950"

0 commit comments

Comments
 (0)