add dl support 90a

4e9fdbe5 · wangshaojie6 · 6b870bc0 · 4e9fdbe5
Commit 4e9fdbe5 authored Jun 10, 2022 by wangshaojie6
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

include/ck/tensor_operation/gpu/device/device_gemm_dl.hpp include/ck/tensor_operation/gpu/device/device_gemm_dl.hpp +1 -1

No files found.
--- a/include/ck/tensor_operation/gpu/device/device_gemm_dl.hpp
+++ b/include/ck/tensor_operation/gpu/device/device_gemm_dl.hpp
@@ -473,7 +473,7 @@ struct DeviceGemmDl

    static bool IsSupportedArgument(const Argument& arg)
    {
-        if(ck::get_device_name() == "gfx906" || ck::get_device_name() == "gfx1030")
+        if(ck::get_device_name() == "gfx906" || ck::get_device_name() == "gfx1030" || ck::get_device_name() == "gfx90a")
        {
            return GridwiseGemm::CheckValidity(
                arg.a_grid_desc_k0_m_k1_, arg.b_grid_desc_k0_n_k1_, arg.c_grid_desc_m_n_);