Fixed contraction issues (#960)

* add missing ComputeType * fixed * Update cmake-ck-dev.sh --------- Co-authored-by: Jing Zhang <jizha@amd.com>

Fixed contraction issues (#960)
* add missing ComputeType * fixed * Update cmake-ck-dev.sh --------- Co-authored-by: Jing Zhang <jizha@amd.com>
aa46039f · zjing14 · GitHub · f477fca4 · aa46039f · aa46039f
Unverified Commit aa46039f authored Oct 03, 2023 by zjing14 Committed by GitHub Oct 03, 2023
2 changed files
--- a/example/61_contraction_multi_ABD/contraction_multi_ABD_xdl_fp16.cpp
+++ b/example/61_contraction_multi_ABD/contraction_multi_ABD_xdl_fp16.cpp
@@ -291,6 +291,7 @@ int main(int argc, char* argv[])
                                                                      BDataType,
                                                                      CShuffleDataType,
                                                                      AccDataType,
+                                                                      A0DataType,
                                                                      PassThrough,
                                                                      BElementOp>;

--- a/include/ck/tensor_operation/gpu/grid/gridwise_gemm_multiple_abd_xdl_cshuffle.hpp
+++ b/include/ck/tensor_operation/gpu/grid/gridwise_gemm_multiple_abd_xdl_cshuffle.hpp
@@ -657,6 +657,7 @@ struct GridwiseGemmMultipleABD_xdl_cshuffle
        auto blockwise_gemm = BlockwiseGemmXdlops_k0mk1_k0nk1_m0n0m1n1m2m3m4n2_Selector<
            BlockSize,
            ComputeDataType,
+            ComputeDataType,
            AccDataType,
            decltype(a_block_desc_ak0_m_ak1),
            decltype(b_block_desc_bk0_n_bk1),