Pat Vignola 213eea216a [MTIA] Add _mtia_maybeExchangeDevice to MTIA module (#149340) ... Summary: The FlexAttention path uses `_maybe_exchange_device`, so it will be needed eventually for MTIA as well. Test Plan: `buck2 test fbcode//mtia/host_runtime/torch_mtia/tests:test_torch_mtia_api -- test_maybe_exchange_device` Reviewed By: chaos5958 Differential Revision: D70072063 Pull Request resolved: https://github.com/pytorch/pytorch/pull/149340 Approved by: https://github.com/chaos5958		2025-03-18 15:15:12 +00:00
..
__init__.py	[MTIA] Add _mtia_maybeExchangeDevice to MTIA module (#149340)	2025-03-18 15:15:12 +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