Signed-off-by: phaedonsun <phaedonsun@tencent.com> Co-authored-by: phaedonsun <phaedonsun@tencent.com>
with torch.cuda.device
with torch.accelerator.device_index
torch.cuda.synchronize()
torch.accelerator.synchronize
Optional[x]
x | None
Union[x, y]
x | y
mypy