mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
Adds create_graph support if you don't compile or compile only with torch.compile(backend="eager"). Using a backend that uses AOTDispatch produces a post-dispatch AOT backward, where its double backward will be silently incorrect if the forward trace involved any ops that are not composite implicit. Pull Request resolved: https://github.com/pytorch/pytorch/pull/153222 Approved by: https://github.com/jansel ghstack dependencies: #153193 |
||
|---|---|---|
| .. | ||
| accumulate_grad.cpp | ||
| accumulate_grad.h | ||
| basic_ops.cpp | ||
| basic_ops.h | ||
| comm.cpp | ||
| comm.h | ||
| init.cpp | ||
| pybind.h | ||
| tensor.cpp | ||
| tensor.h | ||
| utils.cpp | ||
| utils.h | ||