pytorch/torch/mtia
2025-06-17 08:18:47 +00:00
..
__init__.py [BE][PYFMT] migrate PYFMT for torch/[e-n]*/ to ruff format (#144553) 2025-06-17 08:18:47 +00:00
_utils.py torch.mtia module for MTIA device backend (#123612) 2024-04-26 16:17:54 +00:00
memory.py [MTIA] (4/n) Implement PyTorch APIs to query/reset device peak memory usage (#146751) 2025-02-11 03:51:48 +00:00