pytorch
ba8d5f6f - [JIT] FuseLinear pass now handles CallFunction("linear", ...) (#61646)

Commit
2 years ago
[JIT] FuseLinear pass now handles CallFunction("linear", ...) (#61646) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/61646 There are several passes which are written to handle both `CallFunction("linear", ...)` and `aten::linear(...)` despite the two being functionally identical. This changes `FuseLinear` to alse normalize the `CallFunction` variant to `aten::linear`. That way each subsequent transformation only has to handle one form instead of both. Test Plan: Imported from OSS Reviewed By: mikaylagawarecki Differential Revision: D33754261 Pulled By: albanD fbshipit-source-id: 42465cea790538481efc881a249dafdda4bba5d4 (cherry picked from commit ebeca9434caf74c5e75f61b98db443779fe5c6a9)
Author
Committer
Parents
Loading