pytorch

2b6d983b - Reland [dynamo][activation checkpointing] Trace through ActivationWrapper (#109327)

Reland [dynamo][activation checkpointing] Trace through ActivationWrapper (#109327) Fixes https://github.com/pytorch/pytorch/issues/108269 Original reverted PR - https://github.com/pytorch/pytorch/pull/108599 Pull Request resolved: https://github.com/pytorch/pytorch/pull/109327 Approved by: https://github.com/aakhundov