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 пока не охвачены.