pytorch
1f0dcf39 - [JIT] dont optimize device dtype on inline (#43363)

Commit
5 years ago
[JIT] dont optimize device dtype on inline (#43363) Summary: Follow up to https://github.com/pytorch/pytorch/pull/36404 Adding prim::device and prim::dtype to list of skipped peepholes when we run inlining. In the long term another fix may not be to encode shape / dtype info on the traced graph, because it is not guaranteed to be correct. This is blocked by ONNX currently. Partial fix for https://github.com/pytorch/pytorch/issues/43134 Pull Request resolved: https://github.com/pytorch/pytorch/pull/43363 Reviewed By: glaringlee Differential Revision: D23383987 Pulled By: eellison fbshipit-source-id: 2e9c5160d39d690046bd9904be979d58af8d3a20
Author
Elias Ellison
Parents
Loading