reset_max_memory_reserved

paddle.device. reset_max_memory_reserved ( device=None ) [源代码] 

重置给定设备上由内存分配器管理的内存峰值统计。

参数

device (paddle.CUDAPlace|paddle.CustomPlace|paddle.XPUPlace|str|int，可选) - 设备、设备 ID 或形如 gpu:x、xpu:x 或自定义设备名称的设备字符串。如果为 None，则重置当前设备的统计信息。默认值为 None。

返回

None

代码示例

          >>> import paddle
>>> paddle.device.reset_max_memory_reserved('npu:0')
>>> paddle.device.reset_max_memory_reserved('npu')
>>> paddle.device.reset_max_memory_reserved(0)
>>> paddle.device.reset_max_memory_reserved(Paddle.CustomPlace('npu',0))

         