pytorch
46e7f677 - [Static Runtime] Check for inplace ops explicitly in ReplaceWithCopy (#54657)

Commit
3 years ago
[Static Runtime] Check for inplace ops explicitly in ReplaceWithCopy (#54657) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/54657 The constraint checked in D27145406 (https://github.com/pytorch/pytorch/commit/acf03b13f157f8f0190d7ef70591490a2e8585b4) is too tight for the adindexer model and as a result, 5 ops (4 aten::narrow + 1 aten::premute) are not replaced with the copy version and resulted in perf regression. This diff checks for inplace ops explicitly and only applies the input constraint to graphs with inplace ops. Test Plan: Contbuild Reviewed By: ajyu Differential Revision: D27253145 fbshipit-source-id: 23e2b1a018c84dd0fc2880fddd9c41bc0422b8eb
Author
Hao Lu
Parents
Loading