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

Method find_op

dnn/src/cuda/cutlass/operation_table.cpp:154–162  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

152/////////////////////////////////////////////////////////////////////////////////////////////////
153
154Operation const* OperationTable::find_op(GemmKey const& key) const {
155 if (gemm_operations.count(key)) {
156 auto const& ops = gemm_operations.at(key);
157 megdnn_assert(
158 ops.size() == 1, "exactly one kernel expected, got %zu", ops.size());
159 return ops[0];
160 }
161 return nullptr;
162}
163
164/////////////////////////////////////////////////////////////////////////////////////////////////
165

Callers 11

get_available_opMethod · 0.80
get_available_opMethod · 0.80
get_available_opMethod · 0.80
get_available_opMethod · 0.80
get_available_opMethod · 0.80
get_available_opMethod · 0.80
get_available_opMethod · 0.80
execMethod · 0.80
get_available_opMethod · 0.80
get_available_opMethod · 0.80
get_cutlass_conv_opMethod · 0.80

Calls 3

countMethod · 0.45
atMethod · 0.45
sizeMethod · 0.45

Tested by

no test coverage detected