From 79480f0aee29e4b3362d968447547ee6f63a8435 Mon Sep 17 00:00:00 2001
From: "Po-Yen, Chen" <PoYen.Chen@amd.com>
Date: Fri, 19 Aug 2022 01:39:46 -0400
Subject: [PATCH] Re-use element-wise operation type alias

---
 example/01_gemm/gemm_xdl_int4.cpp | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/example/01_gemm/gemm_xdl_int4.cpp b/example/01_gemm/gemm_xdl_int4.cpp
index 0028ba93e8..524d0f4ef8 100644
--- a/example/01_gemm/gemm_xdl_int4.cpp
+++ b/example/01_gemm/gemm_xdl_int4.cpp
@@ -150,9 +150,9 @@ int main(int argc, char* argv[])
     a_m_k_device_buf.ToDevice(a_m_k.mData.data());
     b_k_n_device_buf.ToDevice(b_k_n.mData.data());
 
-    auto a_element_op = PassThrough{};
-    auto b_element_op = PassThrough{};
-    auto c_element_op = PassThrough{};
+    auto a_element_op = AElemenetOp{};
+    auto b_element_op = BElemenetOp{};
+    auto c_element_op = CElemenetOp{};
 
     // do GEMM
     auto gemm    = DeviceGemmInstance{};