diff --git a/include/ck/tensor_operation/gpu/grid/gridwise_gemm_xdl_cshuffle_v3.hpp b/include/ck/tensor_operation/gpu/grid/gridwise_gemm_xdl_cshuffle_v3.hpp index 68c8006a15..4ce2976524 100644 --- a/include/ck/tensor_operation/gpu/grid/gridwise_gemm_xdl_cshuffle_v3.hpp +++ b/include/ck/tensor_operation/gpu/grid/gridwise_gemm_xdl_cshuffle_v3.hpp @@ -1375,6 +1375,7 @@ struct GridwiseGemm_xdl_cshuffle_v3 { #if defined(__gfx950__) return + !DoElementwiseBeforeCShuffle && std::is_same_v && std::is_same_v; #else @@ -1651,9 +1652,7 @@ struct GridwiseGemm_xdl_cshuffle_v3 CShuffleDataType, decltype(c_thread_desc_m0_n0_m1_n1_m2_m3_m4_n2), decltype(c_block_desc_m0_n0_m1_n1_m2_m3_m4_n2), - conditional_t, + tensor_operation::element_wise::PassThrough, Sequence, Sequence