set stream everytime when we get a cuDNN handle (#31541)
Summary:
cudnn version of https://github.com/pytorch/pytorch/pull/31537
https://github.com/pytorch/pytorch/pull/31532 is a quick fix and this is a bigger change. This would deprecate https://github.com/pytorch/pytorch/pull/31532, but we could also merge https://github.com/pytorch/pytorch/pull/31532 first for a quick fix and then work on this later.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/31541
Differential Revision: D19206753
Pulled By: ngimel
fbshipit-source-id: 3352f923d13a9baf0971f64f8b7ce03e9a8b42b1