torch.mtia.memory
Создано: 09 дек. 2024 | Последнее обновление: 06 мая 2026
Бэкенд MTIA реализован вне основного дерева исходного кода; здесь определены только интерфейсы.
Этот пакет добавляет поддержку управления памятью устройства, реализованную в MTIA.
reset_peak_memory_stats
| Сбросить статистику пикового потребления памяти для указанного устройства. |
max_memory_allocated
| Вернуть максимальный объём выделенной памяти в байтах для указанного устройства. |
memory_stats
| Вернуть словарь со статистикой распределителя памяти MTIA для указанного устройства. |
memory_allocated
| Вернуть текущий объём памяти MTIA, занимаемый тензорами, в байтах для указанного устройства. |
© 2026, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://docs.pytorch.org/docs/2.14/mtia.memory.html
