MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / exec

Method exec

dnn/src/rocm/batched_matrix_mul/opr_impl.cpp:47–54  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

45}
46
47void BatchedMatrixMulForwardImpl::exec(
48 _megdnn_tensor_in A, _megdnn_tensor_in B, _megdnn_tensor_out C,
49 _megdnn_workspace workspace) {
50 check_exec(A.layout, B.layout, C.layout, workspace.size);
51 AlgoBase::ExecArgs args(this, A, B, C, workspace);
52 auto&& algo = get_algorithm(this, A.layout, B.layout, C.layout);
53 algo->exec(args);
54}
55
56// vim: syntax=cpp.doxygen

Callers

nothing calls this directly

Calls 1

get_algorithmFunction · 0.85

Tested by

no test coverage detected