pytorch
b4ba66ae - Print tensor shapes and convolution parameters when cuDNN exception is thrown (#45023)

Commit
4 years ago
Print tensor shapes and convolution parameters when cuDNN exception is thrown (#45023) Summary: Originally proposed at https://github.com/pytorch/pytorch/issues/44473#issuecomment-690670989 by colesbury . This PR adds the functionality to print relevant tensor shapes and convolution parameters along with the stack trace once a cuDNN exception is thrown. Pull Request resolved: https://github.com/pytorch/pytorch/pull/45023 Reviewed By: gchanan Differential Revision: D23932661 Pulled By: ezyang fbshipit-source-id: 5f5f570df6583271049dfc916fac36695f415331
Author
Parents
Loading