torch.mtia.memory

创建于：2024年12月09日 | 最后更新于：2025年10月29日

MTIA 后端是树外实现的，此处仅定义接口。

此包添加了对 MTIA 实现的设备内存管理的支持。

memory_stats	返回给定设备的 MTIA 内存分配器统计信息的字典。
memory_allocated	返回给定设备上张量占用的当前 MTIA 内存，单位为字节。