Unify MakeArgument() usage

ceebf306 · Po-Yen, Chen · 21ed2ced · ceebf306 · ceebf306
Commit ceebf306 authored May 05, 2023 by Po-Yen, Chen
Showing with 8 additions and 2 deletions

example/01_gemm/run_gemm_example.inc example/01_gemm/run_gemm_example.inc +4 -1

include/ck/tensor_operation/gpu/device/impl/device_gemm_xdl_cshuffle.hpp ...or_operation/gpu/device/impl/device_gemm_xdl_cshuffle.hpp +4 -1

No files found.
--- a/example/01_gemm/run_gemm_example.inc
+++ b/example/01_gemm/run_gemm_example.inc
@@ -89,7 +89,10 @@ bool run_gemm(const ProblemSize& problem_size, const ExecutionConfig& config)
        K,
        StrideA,
        StrideB,
-        StrideC);
+        StrideC,
+        a_element_op,
+        b_element_op,
+        c_element_op);

    if(!gemm.IsSupportedArgument(argument))
    {

--- a/include/ck/tensor_operation/gpu/device/impl/device_gemm_xdl_cshuffle.hpp
+++ b/include/ck/tensor_operation/gpu/device/impl/device_gemm_xdl_cshuffle.hpp
@@ -207,7 +207,10 @@ struct DeviceGemm_Xdl_CShuffle : public DeviceGemm<ALayout,
                             index_t K,
                             index_t StrideA,
                             index_t StrideB,
-                             index_t StrideC)
+                             index_t StrideC,
+                             AElementwiseOperation,
+                             BElementwiseOperation,
+                             CElementwiseOperation)
    {
        return Argument{p_a,
                        p_b,