pytorch
39470387 - [JIT] enable NNC cpu fusion with torch.jit.fuser("fuser1") (#74078)

Commit
2 years ago
[JIT] enable NNC cpu fusion with torch.jit.fuser("fuser1") (#74078) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/74078 torch.jit.fuser("fuser1") is supposed to enable NNC fusion, but it currently only enables gpu fusion. This will enable CPU fusion as well. Test Plan: Imported from OSS Reviewed By: huiguoo Differential Revision: D34808602 Pulled By: davidberard98 fbshipit-source-id: 2ba26220eb5dbd15c17e57274686961f3bbf46fd (cherry picked from commit c2aa8c187d081f6033e937dd4c2bd579f251b5a0)
Author
Committer
Parents
Loading