Signed-off-by: ilmarkov <markovilya197@gmail.com>
torch.cuda.synchronize()
torch.accelerator.synchronize