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

Method get_workspace_in_bytes

dnn/src/naive/svd/opr_impl.cpp:268–272  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

266namespace naive {
267
268size_t SVDForwardImpl::get_workspace_in_bytes(
269 size_t block_cnt, size_t m, size_t n, size_t dtype_size) {
270 MEGDNN_MARK_USED_VAR(block_cnt);
271 return get_workspace_bundle(m, n, dtype_size).total_size_in_bytes();
272}
273
274WorkspaceBundle SVDForwardImpl::get_workspace_bundle(
275 size_t m, size_t n, size_t dtype_size, void* raw_ptr) {

Callers

nothing calls this directly

Calls 2

total_size_in_bytesMethod · 0.80
get_workspace_bundleFunction · 0.50

Tested by

no test coverage detected