reset_max_memory_reserved

paddle.device.xpu. reset_max_memory_reserved ( device: _XPUPlaceLike | None = None ) None [source]

Reset 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.

Examples

>>> 
>>> import paddle
>>> paddle.device.set_device('xpu')

>>> paddle.device.xpu.reset_max_memory_reserved(paddle.XPUPlace(0))
>>> paddle.device.xpu.reset_max_memory_reserved(0)
>>> paddle.device.xpu.reset_max_memory_reserved("xpu:0")