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().
torch.cuda.set_sync_debug_mode()