pytorch
856510c9 - [JIT] Dont optimize shape info in batch_mm (#44565)

Commit
5 years ago
[JIT] Dont optimize shape info in batch_mm (#44565) Summary: We run remove profile nodes and specialize types before batch_mm, so we cannot run peepholes on the type information of tensors since these properties have not been guarded to be guaranteed to be correct. Pull Request resolved: https://github.com/pytorch/pytorch/pull/44565 Reviewed By: albanD Differential Revision: D23661538 Pulled By: eellison fbshipit-source-id: 0dd23a65714f047f49b4db4ec582b21870925fe1
Author
Elias Ellison
Parents
Loading