viable/strict/1791914049: [CUDA] Migrate UVM MemPool to Python Allocator (#199639)
- PyTorch 近 90 天出现 2268 次
- PyTorch 近 90 天第 2238 次发版
- 上一次：同一天稍早 · viable/strict/1791910592
发生了什么
Replace existing ctypes based UVM mempool to use the new Python allocator introduced in #199638 and #200427 __ Ran existing tests locally: • python test/test_cuda.py -k use_uvm -v test_use_uvm ... ok test_use_uvm_advice_error_frees_allocation ... ok test_use_uvm_backward ... ok test_use_uvm_numerics ... ok test_use_uvm_oom_uses_caching_allocator_recovery ... ok test_use_uvm_tensor_outlives_context ... ok -----------…
摘要按规则整理自下方来源原文