diff --git 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 index 6bac46b062..8e019c5530 100644 --- 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 @@ -809,7 +809,11 @@ struct DeviceContractionMultipleD_Xdl_CShuffle << AK1 << ", " << BK1 << ", " << ABlockTransferSrcVectorDim << ", " - << BBlockTransferSrcVectorDim + << BBlockTransferSrcVectorDim << ", " + << ABlockTransferSrcScalarPerVector << ", " + << BBlockTransferSrcScalarPerVector << ", " + << ABlockLdsExtraM << ", " + << BBlockLdsExtraN << ">"; // clang-format on