torch.mtia.record_memory_history

torch.mtia.record_memory_history(enabled='all', stacks='python', max_entries=0)[source][source]

启用/禁用 MTIA 分配器上的内存分析器

参数:

• 启用（全部或状态，可选）- 由 current_device() 返回的当前设备统计信息，如果 device 为 None（默认）。

• 栈（"python" 或 "cpp"，可选）

• 最大条目数（int，可选）