Rate this Page
★★★★★

torch.mtia.memory.memory_stats#

torch.mtia.memory.memory_stats(device=None)[source]#

Return a dictionary of MTIA memory allocator statistics for a given device.

Parameters

device (torch.device,str, orint,optional) – statistics for the current device, given by current_device(),if device is None (default).

Return type

dict[str,Any]