Autocast support for scaled_dot_product_attention (#91066)
Summary: Autocast support for scaled_dot_product_attention
Test Plan: sandcastle and guthub cicd
Differential Revision: D42085525
Pull Request resolved: https://github.com/pytorch/pytorch/pull/91066
Approved by: https://github.com/ngimel, https://github.com/drisspg