← Back to events
ActiveAIRelease

trunk/26323201f5d6cc7d7daadc11ca718a4db1d924da: [xpu] Support torch.xpu.mem_get_info on pluggable allocator (#197412)

What happened

Motivation Without this code change, torch.xpu.mem_get_info and torch.accelerator.get_memory_info doesn't support on pluggable allocator. ", line 48, in File "/home/sdp/guangyey/pytorch/torch/xpu/memory.py", line 212, in mem_get_info return torch._C._xpu_getMemoryInfo(device) NotImplementedError: getMemoryInfo is not implemented for this allocator yet."> (guangyey) sdp@emr7470: ~ /guangyey/pytorch$ pytest test/test_…

Summary assembled by rule from the sources below

Why it's spreading

Sources