diff --git a/aten/src/ATen/Context.cpp b/aten/src/ATen/Context.cpp index fefcf731cd825..a1d8b696c50a2 100644 --- a/aten/src/ATen/Context.cpp +++ b/aten/src/ATen/Context.cpp @@ -338,6 +338,9 @@ at::BlasBackend Context::blasPreferredBackend() { #if ROCM_VERSION >= 60400 "gfx1200", "gfx1201", #endif +#if ROCM_VERSION >= 60402 + "gfx1150", "gfx1151", +#endif #if ROCM_VERSION >= 60500 "gfx950" #endif @@ -364,6 +367,9 @@ at::BlasBackend Context::blasPreferredBackend() { #if ROCM_VERSION >= 60300 "gfx1100", "gfx1101", "gfx1200", "gfx1201", #endif +#if ROCM_VERSION >= 60402 + "gfx1150", "gfx1151", +#endif #if ROCM_VERSION >= 60500 "gfx950" #endif diff --git a/aten/src/ATen/native/cuda/Blas.cpp b/aten/src/ATen/native/cuda/Blas.cpp index 1fc9e14189e46..aea89c3d4eb24 100644 --- a/aten/src/ATen/native/cuda/Blas.cpp +++ b/aten/src/ATen/native/cuda/Blas.cpp @@ -272,6 +272,9 @@ static bool isSupportedHipLtROCmArch(int index) { #if ROCM_VERSION >= 60300 "gfx1100", "gfx1101", "gfx1200", "gfx1201", #endif +#if ROCM_VERSION >= 60402 + "gfx1150", "gfx1151", +#endif #if ROCM_VERSION >= 60500 "gfx950" #endif