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

Method mem_node

src/core/impl/comp_node/rocm/comp_node.cpp:353–357  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

351}
352
353MemNode ROCmCompNodeImpl::mem_node() {
354 // m_device_info would be null before async init finishes; so we just return
355 // a prive pointer related to device number here
356 return MemNode{sd->dev_info + m_locator.device};
357}
358
359void ROCmCompNodeImpl::sync() {
360 activate();

Callers 15

load_tensor_sharedMethod · 0.45
load_tensor_sharedMethod · 0.45
TESTFunction · 0.45
TESTFunction · 0.45
tensor.cppFile · 0.45
make_proxyMethod · 0.45
var_node.cppFile · 0.45
fwd_in2out_readonlyMethod · 0.45

Calls

no outgoing calls

Tested by 4

TESTFunction · 0.36
TESTFunction · 0.36
TESTFunction · 0.36