Signed-off-by: baoloongmao <baoloongmao@tencent.com> Co-authored-by: Yihua Cheng <yihua98@uchicago.edu>
torch.cuda.synchronize()
torch.accelerator.synchronize
Optional[x]
x | None
Union[x, y]
x | y
mypy