<ЦiSrSSKJrJrJr SSKrSSKJrJr SSK J r S S\\S\\ \44S jjr S S\\S\ 4S jjrS S /rg)zKThis package adds support for device memory management implemented in MTIA.)AnyDictOptionalN) _device_tis_initialized)_get_device_indexdevicereturncr[5(d0$[RR[ USS95$)aReturn a dictionary of MTIA memory allocator statistics for a given device. Args: device (torch.device, str, or int, optional) selected device. Returns statistics for the current device, given by current_device(), if device is None (default). T)optional)rtorch_C_mtia_memoryStatsr r s P/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/torch/mtia/memory.py memory_statsr s/    88 % %&7&N OOc4[US9RSS5$)aReturn the maximum memory allocated in bytes for a given device. Args: device (torch.device or int, optional): selected device. Returns statistic for the current device, given by :func:`~torch.mtia.current_device`, if :attr:`device` is ``None`` (default). rzallocated_bytes.all.peakr)rgetrs rmax_memory_allocatedrs v & * *+Eq IIrrr)N)__doc__typingrrrrrr_utilsr strrintr__all__rrr sgS&& '% P), PS#X P J)!4 J J r