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

Function get_reserved_memory

imperative/python/megengine/device.py:199–207  ·  view source on GitHub ↗

r"""Returns the current memory managed by the caching allocator on the computing device in bytes. Due to the asynchronous execution of MegEngine, please call megengine._full_sync before calling this function in order to get accurate value.

(device: Optional[str] = None)

Source from the content-addressed store, hash-verified

197
198
199def get_reserved_memory(device: Optional[str] = None):
200 r"""Returns the current memory managed by the caching allocator on the computing device in bytes.
201
202 Due to the asynchronous execution of MegEngine, please call megengine._full_sync
203 before calling this function in order to get accurate value.
204 """
205 if device is None:
206 device = get_default_device()
207 return CompNode(device).get_reserved_memory
208
209
210def get_max_reserved_memory(device: Optional[str] = None):

Callers

nothing calls this directly

Calls 2

CompNodeClass · 0.85
get_default_deviceFunction · 0.70

Tested by

no test coverage detected