torch.cuda.mem_get_info
torch.accelerator.get_memory_info
Signed-off-by: Kunshang Ji <kunshang.ji@intel.com> Signed-off-by: Kunshang Ji <jikunshang95@gmail.com>
Signed-off-by: Yan Ma <yan.ma@intel.com>