Frost Mitchell db01f1032f Support XPU in memory tracker (#150703) ...

This PR adds support for XPU devices to the distributed MemoryTracker tool, including unit test for XPU.

Specifically, this code adds tracking for a few alloc-related statistics for XPUCachingAllocator. It also adapts the existing memory tracker tool to be device agnostic, by getting the device module and recording the necessary memory stats. (I get the device module instead of using `torch.accelerator` methods, as that API is still in-progress.)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/150703
Approved by: https://github.com/EikanWang, https://github.com/guangyey, https://github.com/gujinghui, https://github.com/d4l3k

2025-06-12 21:33:52 +00:00

..

memory_tracker_example.py

Support XPU in memory tracker (#150703)

2025-06-12 21:33:52 +00:00