test_async_scheduling.py
Signed-off-by: Nick Hill <nickhill123@gmail.com>
torch.cuda.empty_cache
torch.accelerator.empty_cache