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

Method get_algorithm_heuristic

dnn/src/naive/local_share/opr_impl.cpp:161–168  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

159}
160
161LocalShareForward::Algorithm* LocalShareForwardImpl::get_algorithm_heuristic(
162 const TensorLayout& /* src */, const TensorLayout& /* diff */,
163 const TensorLayout& /* grad */, size_t /* workspace_limit_in_bytes */,
164 const AlgoAttribute& positive_attr, const AlgoAttribute& negative_attr) {
165 auto algo = static_cast<HandleImpl*>(handle())->default_local_share_fwd_algo();
166 algo->check_attribute(positive_attr, negative_attr);
167 return algo;
168}
169
170LocalShareForward::Algorithm* LocalShareForwardImpl::get_algorithm_from_desc(
171 const AlgorithmDesc& desc) {

Callers

nothing calls this directly

Tested by

no test coverage detected