Signed-off-by: Chaojun Zhang <chaojun.zhang@intel.com>
torch.cuda.mem_get_info
torch.accelerator.get_memory_info