with torch.cuda.device
with torch.accelerator.device_index
Signed-off-by: Yan Ma <yan.ma@intel.com>
aiter
amd-aiter