mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
We should use no_reentrant. There are a lot of users of this API, but it is in a prototype state so should be fine to change. Differential Revision: [D48898148](https://our.internmc.facebook.com/intern/diff/D48898148/) Pull Request resolved: https://github.com/pytorch/pytorch/pull/108435 Approved by: https://github.com/awgu ghstack dependencies: #108032, #108033 |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| checkpoint_wrapper.py | ||