Support fp64 contraction on gfx94x. (#1029)

* enable contraction fp64 on gfx94* * fix the logic

Support fp64 contraction on gfx94x. (#1029)
* enable contraction fp64 on gfx94* * fix the logic
ce526211 · Illia Silin · GitHub · 98fd41f5 · ce526211
Unverified Commit ce526211 authored Nov 08, 2023 by Illia Silin Committed by GitHub Nov 08, 2023
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;
        }