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

Function get_max_reserved_memory

imperative/python/megengine/device.py:210–218  ·  view source on GitHub ↗

r"""Returns the maximum 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

208
209
210def get_max_reserved_memory(device: Optional[str] = None):
211 r"""Returns the maximum memory managed by the caching allocator on the computing device in bytes.
212
213 Due to the asynchronous execution of MegEngine, please call megengine._full_sync
214 before calling this function in order to get accurate value.
215 """
216 if device is None:
217 device = get_default_device()
218 return CompNode(device).get_max_reserved_memory
219
220
221def get_max_allocated_memory(device: Optional[str] = None):

Callers 1

compileMethod · 0.85

Calls 2

CompNodeClass · 0.85
get_default_deviceFunction · 0.70

Tested by

no test coverage detected