torch.mtia.record_memory_history

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

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

参数

• enabled (all 或 state, 可选) – 当前设备的统计信息，如果 device 为 None，则由 current_device() 提供（默认）。

• stacks ("python" 或 "cpp", 可选) –

• max_entries (int, 可选) –