Signed-off-by: Sage Moore <sage@neuralmagic.com> Co-authored-by: Tyler Michael Smith <tyler@neuralmagic.com>
torch.cuda.synchronize()
torch.accelerator.synchronize