pytorch
2fd6c33b - [cuDNN] Change `cuDNN` context manager to enable cuDNN as default

Commit
1 year ago
[cuDNN] Change `cuDNN` context manager to enable cuDNN as default It seems a little nonsensical to have the `cuDNN` context manager disable `cuDNN` by default, which would make setting the remaining flags useless. See https://github.com/pytorch/pytorch/blob/7d25a4125180732e8e6882c898bcadd9c9c637a3/test/test_nn.py#L2104 for an example of a nonsense usage. Will wait for thorough CI signal in case we uncover real unexpected breakages. CC @ptrblck @xwang233 @ngimel
References
Author
eqy eqy
Parents
Loading