disable tensor contraction f64 on MI100 (#602)

2026-05-03 21:21:22 +00:00 · 2023-02-23 18:59:37 -06:00
parent 246ceee49e
commit 209baee299
2 changed files with 7 additions and 2 deletions
--- 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
@@ -586,6 +586,11 @@ struct DeviceContractionMultipleD_Xdl_CShuffle
            return false;
        }

+        if(ck::get_device_name() != "gfx90a" && std::is_same<ADataType, double>::value)
+        {
+            return false;
+        }
+
        if(!GridwiseGemm::CheckValidity(arg.a_grid_desc_m_k_,
                                        arg.b_grid_desc_n_k_,
                                        arg.ds_grid_desc_m_n_,