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

Method get_workspace_in_bytes

dnn/src/cambricon/convolution/forward/algo.cpp:115–120  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

113}
114
115size_t ConvolutionForwardImpl::AlgoDefault::get_workspace_in_bytes(
116 const SizeArgs& args) const {
117 auto config = prepare_sub_opr(args);
118 auto bundle = get_workspace_bundle(config, args);
119 return bundle.total_size_in_bytes();
120}
121
122void ConvolutionForwardImpl::AlgoDefault::exec(const ExecArgs& args) const {
123 auto config = prepare_sub_opr(args);

Callers 1

get_workspace_bundleFunction · 0.45

Calls 3

total_size_in_bytesMethod · 0.80
prepare_sub_oprFunction · 0.70
get_workspace_bundleFunction · 0.70

Tested by

no test coverage detected