Elias Ellison
f8a2efc190
Make fusion strategy api public ( #72639 )
...
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/72639
Test Plan: Imported from OSS
Reviewed By: soulitzer
Differential Revision: D34159123
Pulled By: eellison
fbshipit-source-id: 27e4d9694a83e8d6829009882715be4308c96a9f
(cherry picked from commit 1cadcd2f75
2022-02-16 03:45:15 +00:00
Elias Ellison
e0a0f37a11
Add docs for fusion strategy ( #72036 )
...
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/72036
Test Plan: Imported from OSS
Reviewed By: navahgar
Differential Revision: D33864651
Pulled By: eellison
fbshipit-source-id: b1010d97d95fd01603cc7e04a978857cbde3c4fb
(cherry picked from commit e83b008017
2022-02-01 19:07:02 +00:00
jjsjann123
d609957c95
patching graph_for ( #55139 )
...
Summary:
Allows individual DifferentiableGraphOp to display optimized forward graph. This improves user visibility to graph mutation via optimization pass, especially fusion.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/55139
Reviewed By: albanD
Differential Revision: D31330909
Pulled By: dzhulgakov
fbshipit-source-id: c745b482fdc34876dc404cbe3bacd99dcf2ac724
2021-10-04 21:50:22 -07:00
jiej
4d94ee566e
Ge v1 ( #52136 )
...
Summary:
This is a second attempt to use graph executor to run forward on a gradient. This allows a secondary chance to profile intermediate tensor introduced by autodiff.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/52136
Reviewed By: pbelevich
Differential Revision: D26693978
Pulled By: Krovatkin
fbshipit-source-id: 91dde8009a210950af8e5173668ada241e16dd52
2021-02-28 00:53:13 -08:00
Bert Maher
33d51a9b32
Respect canFuseOn{CPU,GPU} in TE fuser ( #43967 )
...
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/43967
Test Plan: Imported from OSS
Reviewed By: asuhan
Differential Revision: D23469048
Pulled By: bertmaher
fbshipit-source-id: 1005a7ae08974059ff9d467492caa3a388070eeb
2020-09-02 18:00:25 -07:00
Michael Suo
ca1b8ebbcb
move misc implementation out of jit/__init__.py ( #41154 )
...
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/41154
Test Plan: Imported from OSS
Reviewed By: ailzhang
Differential Revision: D22445213
Pulled By: suo
fbshipit-source-id: 200545715c5ef13beb1437f49e01efb21498ddb7
2020-07-13 16:59:55 -07:00
