enable contraction fp64 on gfx94*

4f6593af · illsilin · 98fd41f5 · 4f6593af
Commit 4f6593af authored Nov 07, 2023 by illsilin
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

include/ck/tensor_operation/gpu/device/impl/device_contraction_multiple_d_xdl_cshuffle.hpp ...evice/impl/device_contraction_multiple_d_xdl_cshuffle.hpp +3 -1

No files found.
--- a/include/ck/tensor_operation/gpu/device/impl/device_contraction_multiple_d_xdl_cshuffle.hpp
+++ b/include/ck/tensor_operation/gpu/device/impl/device_contraction_multiple_d_xdl_cshuffle.hpp
@@ -595,7 +595,9 @@ struct DeviceContractionMultipleD_Xdl_CShuffle
            return false;
        }

-        if(ck::get_device_name() != "gfx90a" && std::is_same<ADataType, double>::value)
+        if((ck::get_device_name() != "gfx90a" || ck::get_device_name() != "gfx940" ||
+            ck::get_device_name() != "gfx941" || ck::get_device_name() != "gfx942") &&
+           std::is_same<ADataType, double>::value)
        {
            return false;
        }