torch.mtia.record_memory_history
-
torch.mtia.record_memory_history(enabled='all', stacks='python', max_entries=0)[исходный код] -
Включить/отключить профилировщик памяти для распределителя памяти MTIA
- Параметры:
-
- enabled (all or state, необязательно) – статистика для текущего устройства, определяемого current_device(), если device равен None (по умолчанию).
- stacks ("python" or "cpp", необязательно) –
- max_entries (int, необязательно) –
© 2026, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://docs.pytorch.org/docs/2.14/generated/torch.mtia.record_memory_history.html
