Improve documentation for get_memory_info. This feature is lightly defined in PyTorchXLA documentation page. Please provide an explanation on what details it pulls and potentially offer examples.
Additionally, it's important to draw a documentation that clarifies how get_memory_info API works such that users can easily compare/contrast it against torch.cuda.mem_get_info
📚 Documentation
Improve documentation for
get_memory_info. This feature is lightly defined in PyTorchXLA documentation page. Please provide an explanation on what details it pulls and potentially offer examples.Additionally, it's important to draw a documentation that clarifies how
get_memory_infoAPI works such that users can easily compare/contrast it againsttorch.cuda.mem_get_infocc @mikegre-google to help follow up @JackCaoG