pytorch

e01b07e1 - [ROCm] Autocast RNN Support (#121539)

[ROCm] Autocast RNN Support (#121539) Fixes #116361 Implements Autocast wrapper for miopen rnn's Pull Request resolved: https://github.com/pytorch/pytorch/pull/121539 Approved by: https://github.com/albanD, https://github.com/jeffdaily