diff --git a/aten/src/ATen/Context.cpp b/aten/src/ATen/Context.cpp index a1d8b696c50a..cb7971bdc193 100644 --- a/aten/src/ATen/Context.cpp +++ b/aten/src/ATen/Context.cpp @@ -336,7 +336,10 @@ at::BlasBackend Context::blasPreferredBackend() { static const std::vector archs = { "gfx90a", "gfx942", #if ROCM_VERSION >= 60400 - "gfx1200", "gfx1201", + "gfx1100", "gfx1101", "gfx1102", "gfx1200", "gfx1201", +#endif +#if ROCM_VERSION >= 60402 + "gfx1150", "gfx1151", #endif #if ROCM_VERSION >= 60402 "gfx1150", "gfx1151", @@ -365,7 +368,10 @@ at::BlasBackend Context::blasPreferredBackend() { static const std::vector archs = { "gfx90a", "gfx942", #if ROCM_VERSION >= 60300 - "gfx1100", "gfx1101", "gfx1200", "gfx1201", + "gfx1100", "gfx1101", "gfx1102", "gfx1200", "gfx1201", +#endif +#if ROCM_VERSION >= 60402 + "gfx1150", "gfx1151", #endif #if ROCM_VERSION >= 60402 "gfx1150", "gfx1151", diff --git a/aten/src/ATen/native/cuda/Blas.cpp b/aten/src/ATen/native/cuda/Blas.cpp index aea89c3d4eb2..257443e472d5 100644 --- a/aten/src/ATen/native/cuda/Blas.cpp +++ b/aten/src/ATen/native/cuda/Blas.cpp @@ -270,7 +270,10 @@ static bool isSupportedHipLtROCmArch(int index) { static const std::vector archs = { "gfx90a", "gfx942", #if ROCM_VERSION >= 60300 - "gfx1100", "gfx1101", "gfx1200", "gfx1201", + "gfx1100", "gfx1101", "gfx1102", "gfx1200", "gfx1201", +#endif +#if ROCM_VERSION >= 60402 + "gfx1150", "gfx1151", #endif #if ROCM_VERSION >= 60402 "gfx1150", "gfx1151",