From c8e83bab456af35e5bba9295f8e4d934e2c245eb Mon Sep 17 00:00:00 2001 From: zjing14 Date: Tue, 3 Oct 2023 09:32:44 -0500 Subject: [PATCH] Fixed contraction issues (#960) * add missing ComputeType * fixed * Update cmake-ck-dev.sh --------- Co-authored-by: Jing Zhang [ROCm/composable_kernel commit: aa46039f2d25874c799d85f5cd28bc636892deef] --- .../61_contraction_multi_ABD/contraction_multi_ABD_xdl_fp16.cpp | 1 + .../gpu/grid/gridwise_gemm_multiple_abd_xdl_cshuffle.hpp | 1 + 2 files changed, 2 insertions(+) diff --git a/example/61_contraction_multi_ABD/contraction_multi_ABD_xdl_fp16.cpp b/example/61_contraction_multi_ABD/contraction_multi_ABD_xdl_fp16.cpp index 6634dacd25..5a274d485e 100644 --- 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>; diff --git 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 index e7e012c009..6321de73a9 100644 --- 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),