pytorch
69b7b011 - [JIT] Add cond-add-relu matching pattern to cover in-place ops (#55458)

Commit
3 years ago
[JIT] Add cond-add-relu matching pattern to cover in-place ops (#55458) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/55458 Previously the cond-add-relu pass blindly turns all in-place add and relu ops into non-mutation version, when those ops are not part of the fusion patten, it can actually hurt the performance as shown in densenet on some platform. Test Plan: Imported from OSS Reviewed By: eellison Differential Revision: D27620415 fbshipit-source-id: 8302c0c85f3a064dfd8ac994e92416dde927e348
Author
Parents
Loading