torch.cuda.set_sync_debug_mode

torch.cuda.set_sync_debug_mode(debug_mode) [source]

Устанавливает режим отладки для операций синхронизации cuda.

Параметры

debug_mode (str или int) – если “default” или 0, не выдавать ошибки или предупреждения при синхронизирующих операциях, если “warn” или 1, выдавать предупреждение при синхронизирующих операциях, если “error” или 2, выдавать ошибку при синхронизирующих операциях.

Предупреждение

Эта функция находится в стадии разработки, и не все операции синхронизации будут вызывать предупреждение или ошибку. В частности, операции в пространствах имен torch.distributed и torch.sparse пока не поддерживаются.