diff --git a/library/src/jit_library/src/device_gemm_multiple_d.cpp b/library/src/jit_library/src/device_gemm_multiple_d.cpp index 2c381d18fd..29aac50c79 100644 --- a/library/src/jit_library/src/device_gemm_multiple_d.cpp +++ b/library/src/jit_library/src/device_gemm_multiple_d.cpp @@ -38,7 +38,7 @@ std::size_t GetGridSize(const std::size_t m, const std::unordered_set& get_xdlop_archs() { - static std::unordered_set supported_archs{"gfx90a"}; + static std::unordered_set supported_archs{"gfx90a", "gfx908", "gfx940"}; return supported_archs; }