viable/strict/1789741687: Add XPU caching allocator checkpoint pool state bindings (#188259)
- PyTorch 近 90 天出现 1165 次
- PyTorch 近 90 天第 1151 次发版
- 上一次：同一天稍早 · viable/strict/1789737992
发生了什么
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…
摘要按规则整理自下方来源原文