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

Method get_workspace_in_bytes

dnn/src/cuda/checksum/opr_impl.cpp:23–26  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

21} // anonymous namespace
22
23size_t ChecksumForwardImpl::get_workspace_in_bytes(const TensorLayout& data) {
24 auto wbundle = get_wbundle(data);
25 return wbundle.total_size_in_bytes();
26}
27
28ChecksumForward::Result ChecksumForwardImpl::exec(
29 _megdnn_tensor_in data, _megdnn_workspace workspace) {

Callers

nothing calls this directly

Calls 2

total_size_in_bytesMethod · 0.80
get_wbundleFunction · 0.70

Tested by

no test coverage detected