Hyunho Yeo dcac3c3e06 [MTIA] (2/n) Implement PyTorch APIs to query/reset device peak memory usage (#146659) ... Summary: Public summary (shared with Github): This diff implements the correct version of the PyTorch API "max_memory_allocated". Nit: The file previously contained two unit tests with the same name (due to wrong revert); I deleted a deprecated one to revamp the correct version. Test Plan: ``` buck2 test //mtia/host_runtime/torch_mtia/tests:test_torch_mtia_api -- -r test_max_memory_allocated ``` https://www.internalfb.com/intern/testinfra/testrun/12103424065182810 Reviewed By: yuhc Differential Revision: D68988435 Pull Request resolved: https://github.com/pytorch/pytorch/pull/146659 Approved by: https://github.com/nautsimon		2025-02-07 23:06:35 +00:00
..
__init__.py	[S481486] Move MTIA dynamic library loading from __init__.py to a separate module (#145322)	2025-01-22 23:39:43 +00:00
_utils.py
memory.py	[MTIA] (2/n) Implement PyTorch APIs to query/reset device peak memory usage (#146659)	2025-02-07 23:06:35 +00:00