Unverified Commit a0ae1c61 authored by Adam Osewski's avatar Adam Osewski Committed by GitHub
Browse files

Fix MakeArgument (#1284)

parent 3c043cd1
...@@ -92,9 +92,10 @@ bool run_grouped_gemm(const ProblemSize& problem_size, const ExecutionConfig& co ...@@ -92,9 +92,10 @@ bool run_grouped_gemm(const ProblemSize& problem_size, const ExecutionConfig& co
auto group_count = problem_size.group_count; auto group_count = problem_size.group_count;
using KernelArguments = ck::tensor_operation::device::GroupedGemmTileLoopKernelArguments<NumDs>; using KernelArguments = ck::tensor_operation::device::GroupedGemmTileLoopKernelArguments<NumDs>;
using GemmDesc = ck::tensor_operation::device::GemmDesc;
// GEMM shape // GEMM shape
std::vector<ck::tensor_operation::device::GemmDesc> gemm_descs; std::vector<GemmDesc> gemm_descs;
std::vector<KernelArguments> ggemm_kargs; std::vector<KernelArguments> ggemm_kargs;
std::vector<void*> p_Cs; std::vector<void*> p_Cs;
std::vector<const void*> p_As; std::vector<const void*> p_As;
......
...@@ -375,7 +375,7 @@ struct DeviceGroupedGemmMultipleDXdlCShuffleTileLoop ...@@ -375,7 +375,7 @@ struct DeviceGroupedGemmMultipleDXdlCShuffleTileLoop
std::vector<const void*>& /* p_Bs */, std::vector<const void*>& /* p_Bs */,
std::vector<std::array<const void*, NumDTensor>>& /* p_Ds */, std::vector<std::array<const void*, NumDTensor>>& /* p_Ds */,
std::vector<void*>& /* p_Es */, std::vector<void*>& /* p_Es */,
std::vector<GemmDesc>& gemm_descs, const 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,
...@@ -643,7 +643,7 @@ struct DeviceGroupedGemmMultipleDXdlCShuffleTileLoop ...@@ -643,7 +643,7 @@ struct DeviceGroupedGemmMultipleDXdlCShuffleTileLoop
std::vector<const void*>& p_Bs, std::vector<const void*>& p_Bs,
std::vector<std::array<const void*, NumDTensor>>& p_Ds, std::vector<std::array<const void*, NumDTensor>>& p_Ds,
std::vector<void*>& p_Es, std::vector<void*>& p_Es,
std::vector<GemmDesc> gemm_descs, std::vector<GemmDesc>& gemm_descs,
AElementwiseOperation a_elementwise_op, AElementwiseOperation a_elementwise_op,
BElementwiseOperation b_elementwise_op, BElementwiseOperation b_elementwise_op,
CDEElementwiseOperation cde_elementwise_op) CDEElementwiseOperation cde_elementwise_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