Skip to content

GitLab

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

Clean-up grouped GEMM interface.

parent 927de2f4
Hide 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, ...@@ -412,11 +412,11 @@ struct DeviceGroupedGemmMultipleD_Dl : public DeviceGroupedGemm<ALayout,
struct Argument : public BaseArgument struct Argument : public BaseArgument
{ {
Argument(std::vector<const void*, Allocator>& p_As, Argument(std::vector<const void*>& p_As,
std::vector<const void*, Allocator>& p_Bs, std::vector<const void*>& p_Bs,
std::vector<std::array<const void*, NumDTensor>, Allocator>& p_Ds, std::vector<std::array<const void*, NumDTensor>>& p_Ds,
std::vector<void*, Allocator>& p_Es, std::vector<void*>& p_Es,
std::vector<GemmDesc, Allocator>& gemm_descs, std::vector<GemmDesc>& gemm_descs,
AElementwiseOperation a_element_op, AElementwiseOperation a_element_op,
BElementwiseOperation b_element_op, BElementwiseOperation b_element_op,
CDEElementwiseOperation cde_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