pytorch
40fb28ea - [JIT] Compute input sym shapes in partial eval graph (#68281)

Commit
3 years ago
[JIT] Compute input sym shapes in partial eval graph (#68281) Summary: Needed for NNC dynamic shape fusion. Previously, when creating a partially evaluated graph for symbolic shape compute, if the input wasn't used, we wouldn't compute it, which led to failures when NNC expected this value to be passed in. Pull Request resolved: https://github.com/pytorch/pytorch/pull/68281 Reviewed By: navahgar Differential Revision: D32401365 Pulled By: eellison fbshipit-source-id: 97a684e5f1faed5df77c8fd69f9623cdba0781f9
Author
Elias Ellison
Parents
Loading