Unverified Commit e5ec0f39 authored by zjing14's avatar zjing14 Committed by GitHub
Browse files

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


Co-authored-by: default avatarBartłomiej Kocot <barkocot@amd.com>
parent ccf5a65e
...@@ -23,6 +23,24 @@ struct GemmMultiABDDesc ...@@ -23,6 +23,24 @@ struct GemmMultiABDDesc
ck::index_t stride_C_; ck::index_t stride_C_;
}; };
/**
* \brief Grouped Gemm Multi ABD
*
* 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