pytorch
9e59244b - fix view listing in autograd codegen (#32044)

Commit
4 years ago
fix view listing in autograd codegen (#32044) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/32044 Fix the list of views in the codegen: - Move `narrow` out of the autograd functions since it's now implemented with slice. - Add `split_with_sizes` that was missing from the list - Remove special formulas for both `split` and `split_with_sizes`. Both used not to be considered as views. When they are, all the rnn code breaks because it uses them in an invalid way. The generic formula will generate one `narrow` Node for each output. Which is always valid. The diff for the generated code can be found [here](https://github.com/pytorch/pytorch/compare/16eff6e...albanD:06d6e85) (outdated for last commit) Test Plan: Imported from OSS Differential Revision: D19409648 Pulled By: albanD fbshipit-source-id: 5ebc4c978af500403f7f008c0231b7db0cabab26
Author
Parents
Loading