viable/strict/1789741687: Add XPU caching allocator checkpoint pool state bindings (#188259)
- PyTorch: 1159 events in the last 90 days
- PyTorch: 1145th Release in the last 90 days
- Previous: earlier the same day · viable/strict/1789737992
What happened
Adds XPU support for caching allocator private-pool checkpoint state APIs used by graph pool restoration. This change mirrors the existing CUDA private-pool checkpoint plumbing for XPU by: adding XPU allocator checkpoint state and restore helpers exposing _xpu_getCheckpointState , _xpu_setCheckpointPoolState , and _xpu_checkPoolLiveAllocations exposing _xpu_isHistoryEnabled and _xpu_xpuCachingAllocator_raw_delete Pu…
Summary assembled by rule from the sources below