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, 可選) –