Finding synchronization points

Hi,
is there a way (e.g. a ContextManager) to find cpu-gpu synchronization points in my code?#

Best,
Tim

You could use torch.cuda.set_sync_debug_mode().

1 Like