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)
