diff --git a/aten/src/ATen/Context.cpp b/aten/src/ATen/Context.cpp index 01f223f4e5ce7..43875d938e1f5 100644 --- a/aten/src/ATen/Context.cpp +++ b/aten/src/ATen/Context.cpp @@ -335,6 +335,9 @@ at::BlasBackend Context::blasPreferredBackend() { static const bool hipblaslt_preferred = []() { static const std::vector archs = { "gfx90a", "gfx942", +#if ROCM_VERSION >= 60300 + "gfx1200", "gfx1201", +#endif #if ROCM_VERSION >= 60500 "gfx950" #endif