Skip to content

GitLab

Commit af2a1bd8 authored by Ville Pietilä's avatar Ville Pietilä
Browse files

Clean-up grouped GEMM interface.

parent 927de2f4
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 5 deletions
include/ck/tensor_operation/gpu/device/impl/device_grouped_gemm_multiple_d_dl.hpp
View file @ af2a1bd8
......@@ -412,11 +412,11 @@ struct DeviceGroupedGemmMultipleD_Dl : public DeviceGroupedGemm<ALayout,
struct Argument : public BaseArgument
{
Argument(std::vector<const void*, Allocator>& p_As,
std::vector<const void*, Allocator>& p_Bs,
std::vector<std::array<const void*, NumDTensor>, Allocator>& p_Ds,
std::vector<void*, Allocator>& p_Es,
std::vector<GemmDesc, Allocator>& gemm_descs,
Argument(std::vector<const void*>& p_As,
std::vector<const void*>& p_Bs,
std::vector<std::array<const void*, NumDTensor>>& p_Ds,
std::vector<void*>& p_Es,
std::vector<GemmDesc>& gemm_descs,
AElementwiseOperation a_element_op,
BElementwiseOperation b_element_op,
CDEElementwiseOperation cde_element_op)
......
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