+ jRt^RIHt^RIt^RIHtHt^RIHtR RRllt R RRllt R R R llt R R R llt .ROt R#)zKThis package adds support for device memory management implemented in MTIA.)AnyN)Deviceis_initialized)_get_device_indexcRV^8dQhR\R\\\3,/#devicereturn)rdictstrr)formats"R/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/mtia/memory.py __annotate__r s& P P P4S> Pcz\4'g/#\PP\ VRR74#)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). Toptional)rtorch_C_mtia_memoryStatsrr s&r memory_statsr s/    88 % %&7&N OOrc0V^8dQhR\R\/#rrint)r s"rrrs D D D3 Drc|\4'g^#\V4PR^4PR^4#)zReturn the maximum memory allocated in bytes 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). dram peak_bytesrrgetrs&rmax_memory_allocatedr!s5      # #FA . 2 2< CCrc0V^8dQhR\R\/#rr)r s"rrr's I IV Is Irc|\4'g^#\V4PR^4PR^4#)a+Return the current MTIA memory occupied by tensors in bytes for a given device. Args: device (torch.device or int or str, optional): selected device. Returns statistic for the current device, given by :func:`~torch.mtia.current_device`, if :attr:`device` is ``None`` (default). rallocated_bytesrrs&rmemory_allocatedr%'s6      # #FA . 2 23Da HHrc(V^8dQhR\RR/#)rr r N)r)r s"rrr4s R RF Rd Rrc~\4'gR#\PP\ VRR74R#)zReset the peak memory stats 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). NTr)rrr_mtia_resetPeakMemoryStatsrrs&rreset_peak_memory_statsr)4s*    HH''(9&4(PQr)N)rr!r%r))__doc__typingrrrr_utilsrrr!r%r)__all__rrr0s6S $% P D I R r