mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
we use dummy tensors in our initial trace, so we should never inline. the subclass dispatch might not support the dummy tensor, e.g. DTensor accumulate grad will check that both param and grad are DTensors Pull Request resolved: https://github.com/pytorch/pytorch/pull/149014 Approved by: https://github.com/jansel ghstack dependencies: #149064 |
||
|---|---|---|
| .. | ||
| 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 | ||