torch.mtia.memory

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

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

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

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