pytorch
d8277d90 - Disallow saved tensor hooks in functorch transforms (#85829)

Commit
2 years ago
Disallow saved tensor hooks in functorch transforms (#85829) Technically they may only be a problem with the grad transform. Though the branch cut is soon, this is the more conservative change, it also lets us disable checkpointing for functorch (which definitely doesn't work with all transforms) and not a lot of people use saved tensor hooks with functorch (I discovered this while testing). Test Plan: - new tests Pull Request resolved: https://github.com/pytorch/pytorch/pull/85829 Approved by: https://github.com/samdow
Author
Committer
Parents
Loading