Clean-up grouped GEMM interface.

af2a1bd8 · Ville Pietilä · 927de2f4 · af2a1bd8
Commit af2a1bd8 authored Dec 09, 2024 by Ville Pietilä
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 ...ion/gpu/device/impl/device_grouped_gemm_multiple_d_dl.hpp +5 -5

No files found.
--- a/include/ck/tensor_operation/gpu/device/impl/device_grouped_gemm_multiple_d_dl.hpp
+++ b/include/ck/tensor_operation/gpu/device/impl/device_grouped_gemm_multiple_d_dl.hpp
@@ -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)