changed API

include/ck/tensor_operation/gpu/device/device_grouped_gemm_multi_abd.hpp

@@ -55,6 +55,11 @@ struct DeviceGroupedGemmMultiABD : public BaseOperator
                         CElementwiseOperation c_element_op = CElementwiseOperation{}) = 0;
 
     virtual std::unique_ptr<BaseInvoker> MakeInvokerPointer() = 0;
+
+    virtual void SetElementwiseOps(BaseArgument* p_arg,
+                                   AElementwiseOperation a_element_op,
+                                   BElementwiseOperation b_element_op,
+                                   CElementwiseOperation cde_element_op) const = 0;
 };
 
 } // namespace device

include/ck/tensor_operation/gpu/device/device_grouped_gemm_multi_abd_fixed_nk.hpp

@@ -56,10 +56,6 @@ struct DeviceGroupedGemmMultiABDFixedNK : DeviceGroupedGemmMultiABD<AsLayout,
     virtual void SetDeviceKernelArgs(BaseArgument* p_arg, const void* kernel_args) const = 0;
     virtual size_t GetDeviceKernelArgSize(const BaseArgument* p_arg) const               = 0;
     virtual void SetKBatch(BaseArgument* p_arg, index_t k_batch) const                   = 0;
-    virtual void SetElementwiseOps(BaseArgument* p_arg,
-                                   AElementwiseOperation a_element_op,
-                                   BElementwiseOperation b_element_op,
-                                   CElementwiseOperation cde_element_op) const           = 0;
 };
 
 } // namespace device