mindspore.runtime.reset_max_memory_reserved
- mindspore.runtime.reset_max_memory_reserved()[源代码]
重置被内存池所管理的内存的峰值。
样例:
>>> import mindspore as ms >>> import numpy as np >>> from mindspore import Tensor, ops >>> ms.set_device("Ascend", 0) >>> a = Tensor(np.ones([1, 2]), ms.float32) >>> b = Tensor(np.ones([1, 2]), ms.float32) >>> c = ops.add(a, b).asnumpy() >>> print(ms.runtime.max_memory_reserved()) 1073741824 >>> ms.runtime.reset_max_memory_reserved() >>> print(ms.runtime.max_memory_reserved()) 0