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
52b0bffe · Illia Silin · Jun Liu · ebcfdb3b · 52b0bffe
Commit 52b0bffe authored Nov 08, 2023 by Illia Silin Committed by Jun Liu Nov 08, 2023
Show 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;
        }