Signed-off-by: mayuyuace <qiming1.zhang@intel.com>
torch.cuda.mem_get_info
torch.accelerator.get_memory_info