diff --git a/aten/src/ATen/Context.cpp b/aten/src/ATen/Context.cpp index fefcf731cd82..1a679078281e 100644 --- a/aten/src/ATen/Context.cpp +++ b/aten/src/ATen/Context.cpp @@ -336,7 +336,7 @@ at::BlasBackend Context::blasPreferredBackend() { static const std::vector archs = { "gfx90a", "gfx942", #if ROCM_VERSION >= 60400 - "gfx1200", "gfx1201", + "gfx1150", "gfx1151", "gfx1200", "gfx1201", #endif #if ROCM_VERSION >= 60500 "gfx950" diff --git a/aten/src/ATen/native/cuda/Blas.cpp b/aten/src/ATen/native/cuda/Blas.cpp index 1fc9e14189e4..2fc9cdddaf66 100644 --- a/aten/src/ATen/native/cuda/Blas.cpp +++ b/aten/src/ATen/native/cuda/Blas.cpp @@ -270,7 +270,7 @@ static bool isSupportedHipLtROCmArch(int index) { static const std::vector archs = { "gfx90a", "gfx942", #if ROCM_VERSION >= 60300 - "gfx1100", "gfx1101", "gfx1200", "gfx1201", + "gfx1100", "gfx1101", "gfx1150", "gfx1151", "gfx1200", "gfx1201", #endif #if ROCM_VERSION >= 60500 "gfx950"