| 27 | |
| 28 | void exec_int_8x8x16(const MatrixMulImpl::KernParam& kern_param) { |
| 29 | MIDOUT_BEGIN(megdnn_arm_exec_int8816, void) { |
| 30 | auto bundle = get_workspace_bundle_int_8x8x16(kern_param); |
| 31 | bundle.set(kern_param.workspace_ptr); |
| 32 | auto w0 = static_cast<int8_t*>(bundle.get(0)); |
| 33 | auto w1 = static_cast<int8_t*>(bundle.get(1)); |
| 34 | size_t M = kern_param.M; |
| 35 | size_t N = kern_param.N; |
| 36 | size_t K = kern_param.K; |
| 37 | size_t LDB = kern_param.LDB; |
| 38 | exec_gemm_int8_int8_int16( |
| 39 | kern_param.A<dt_int8>(), kern_param.B<dt_int8>(), |
| 40 | kern_param.C<dt_int16>(), M, K, N, LDB, w0, w1); |
| 41 | } |
| 42 | MIDOUT_END(); |
| 43 | } |
| 44 | } // anonymous namespace |
no test coverage detected