"git@developer.sourcefind.cn:modelzoo/resnet50_tensorflow.git" did not exist on "f5014889db3e8f18ea05c40b7954d5fdbabf93e2"
Unverified Commit b92fe98f authored by zjing14's avatar zjing14 Committed by GitHub
Browse files

Update include/ck/tensor_operation/gpu/device/device_grouped_gemm_multi_abd_fixed_nk.hpp


Co-authored-by: default avatarBartłomiej Kocot <barkocot@amd.com>
parent 5a52aa90
...@@ -30,6 +30,24 @@ struct GroupedGemmMultiABDKernelArgument ...@@ -30,6 +30,24 @@ struct GroupedGemmMultiABDKernelArgument
index_t StrideE; index_t StrideE;
}; };
/**
* \brief Grouped Gemm Multi ABD Fixed NK
*
* C = a_op(A, A1...) * b_op(B, B1...)
* E = cde_op(C, D0, D1, ...)
*
* \tparam AsLayout A layouts (tuple).
* \tparam BsLayout B layouts (tuple).
* \tparam DsLayout Ds layouts (tuple).
* \tparam ELayout Output layout.
* \tparam AsDataType A data types (tuple).
* \tparam BsDataType B data types (tuple).
* \tparam DsDataType D data types (tuple).
* \tparam EDataType Output data type.
* \tparam AElementwiseOperation A elementwise operation.
* \tparam BElementwiseOperation B elementwise operation.
* \tparam CDEElementwiseOperation C elementwise operation.
*/
template <typename AsLayout, template <typename AsLayout,
typename BsLayout, typename BsLayout,
typename DsLayout, typename DsLayout,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment