torch.cuda.set_sync_debug_mode

torch.cuda.set_sync_debug_mode(debug_mode)[源代码]

设置CUDA同步操作的调试模式。

参数

debug_mode (str 或 int) – 如果为 "default" 或 0，不对同步操作进行错误或警告；如果为 "warn" 或 1，在同步操作时发出警告；如果为 "error" 或 2，在同步操作时抛出错误。

警告

这是一个实验性功能，并不是所有的同步操作都会触发警告或错误。尤其是 torch.distributed 和 torch.sparse 命名空间中的操作目前还没有被涵盖。