Signed-off-by: Sage Moore <sagmoore@redhat.com> Signed-off-by: Sage Moore <sage@neuralmagic.com>
torch.cuda.synchronize()
torch.accelerator.synchronize