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

Method exec

dnn/src/rocm/convolution/forward/matmul.cpp:23–34  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

21}
22
23void ConvolutionForwardImpl::AlgoMatmul::exec(const ExecArgs& args) const {
24#define cb(DType) \
25 if (args.src_layout->dtype == DType()) { \
26 using ctype = typename DTypeTrait<DType>::ctype; \
27 exec_internal<ctype>(args); \
28 return; \
29 }
30 MEGDNN_FOREACH_COMPUTING_DTYPE_FLOAT(cb)
31#undef cb
32
33 megdnn_assert_internal(0);
34}
35
36template <typename T>
37void ConvolutionForwardImpl::AlgoMatmul::exec_internal(const ExecArgs& args) {

Callers 1

exec_internalMethod · 0.45

Calls

no outgoing calls

Tested by

no test coverage detected