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