max_memory_reserved
- paddle.device.xpu. max_memory_reserved ( device: _XPUPlaceLike | None = None ) int [source]
-
Return the peak size of XPU memory that is held by the allocator of the given device.
- Parameters
-
device (paddle.XPUPlace|int|str|None, optional) – The device, the id of the device or the string name of device like ‘xpu:x’. If device is None, the device is the current device. Default: None.
- Returns
-
The peak size of XPU memory that is held by the allocator of the given device, in bytes.
- Return type
-
int
Examples
>>> >>> import paddle >>> paddle.device.set_device('xpu') >>> max_memory_reserved_size = paddle.device.xpu.max_memory_reserved(paddle.XPUPlace(0)) >>> max_memory_reserved_size = paddle.device.xpu.max_memory_reserved(0) >>> max_memory_reserved_size = paddle.device.xpu.max_memory_reserved("xpu:0")