Use element-wise ops in the reference gemm

library/include/ck/library/reference_tensor_operation/cpu/reference_mx_gemm.hpp

@@ -67,14 +67,13 @@ struct ReferenceMXGemm : public device::BaseOperator
 
         float Run(const Argument& arg)
         {
-            using PassThrough  = ck::tensor_operation::element_wise::PassThrough;
             using GemmInstance = ck::tensor_operation::host::ReferenceGemm<ComputeTypeA,
                                                                            ComputeTypeB,
                                                                            CDataType,
                                                                            AccDataType,
-                                                                           PassThrough,
-                                                                           PassThrough,
-                                                                           PassThrough,
+                                                                           AElementwiseOperation,
+                                                                           BElementwiseOperation,
+                                                                           CElementwiseOperation,
                                                                            ComputeTypeA,
                                                                            ComputeTypeB>;
 
@@ -111,9 +110,9 @@ struct ReferenceMXGemm : public device::BaseOperator
             auto ref_argument = ref_gemm.MakeArgument(a_m_k_scaled,
                                                       b_k_n_scaled,
                                                       arg.c_m_n_,
-                                                      PassThrough{},
-                                                      PassThrough{},
-                                                      PassThrough{});
+                                                      arg.a_element_op_,
+                                                      arg.b_element_op_,
+                                                      arg.c_element_op_);
 
             ref_invoker.Run(ref_argument);