torch.cuda.set_sync_debug_mode# torch.cuda.set_sync_debug_mode(debug_mode)[source]# Set 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.
