• rocking5566's avatar
    Add gemm_shuffle host api (#71) · 4d40b197
    rocking5566 authored
    * [What]
    1. Add DeviceGemmXdl_C_Shuffle
    2. Revise example of gemm_xdl
    [Why] Prepare to add shuffle version of D = alpha * (A * B) + beta * C
    [How] Imitate DeviceGemmXdl and device_conv2d_fwd_xdl_c_shuffle_nhwc_kyxc_nhwk.hpp
    4d40b197
gemm_xdl.cpp 8.95 KB