torch.cuda.set_sync_debug_mode
torch.cuda.set_sync_debug_mode(debug_mode)
Sets the debug mode for cuda synchronizing operations.
Parameters:
      debug_mode (str or int) -- if "default" or 0, don't
      error or warn on synchronizing operations, if "warn" or 1, warn
      on synchronizing operations, if "error" or 2, error out
      synchronizing operations.
Warning:
 This is an experimental feature, and not all synchronizing
 operations will trigger warning or error. In particular,
 operations in torch.distributed and torch.sparse namespaces are
 not covered yet.
