pytorch

cf3ef237 - Propagate full autocast state to CheckpointFunction's forward-inside-backward (#71169)

Propagate full autocast state to CheckpointFunction's forward-inside-backward (#71169) Summary: Should fix https://github.com/pytorch/pytorch/issues/71124 (implements https://github.com/pytorch/pytorch/issues/71124#issuecomment-1009436056). cc mcarilli ptrblck Pull Request resolved: https://github.com/pytorch/pytorch/pull/71169 Reviewed By: albanD Differential Revision: D33793556 Pulled By: ngimel fbshipit-source-id: 80a4b4f0657b922002e3446fb6b48f082fa98453 (cherry picked from commit cf9beee28bf7b541b4631c13fa35bb494212e1cd)