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