fix bfloat16 autocast skip (#67822)

Commit

4 years ago

fix bfloat16 autocast skip (#67822) Summary: Per title Pull Request resolved: https://github.com/pytorch/pytorch/pull/67822 Reviewed By: mruberry Differential Revision: D32162605 Pulled By: ngimel fbshipit-source-id: eb5ccf6c441231e572ec93ac8c2638d028abecad