ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-07 08:15:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

Re-use element-wise operation type alias

Browse Source
This commit is contained in:
Po-Yen, Chen
2022-08-19 01:39:46 -04:00
parent dd849a8736
commit 79480f0aee
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
example/01_gemm/gemm_xdl_int4.cpp
View File

@@ -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{};