James Reed
3f6643e661
[FX] Fix default argument handling for Interpreter ( #72272 )
...
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/72272
Test Plan: Imported from OSS
Reviewed By: dagitses
Differential Revision: D33984862
Pulled By: jamesr66a
fbshipit-source-id: 7d89901c2041806df86c9b08f3af731f3afc9100
(cherry picked from commit f79f0e451e )
2022-02-04 01:46:20 +00:00
James Reed
538647fe1f
[WIP][FX] BC guarantees for 1.10 ( #63888 )
...
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/63888
Test Plan: Imported from OSS
Reviewed By: pbelevich
Differential Revision: D30523133
Pulled By: jamesr66a
fbshipit-source-id: b04cc0d842a74862f42ecba98b757310cd2ec7b0
2021-08-30 19:56:46 -07:00
Jordan Fix
f65793507d
[fx][Transformer] Add override for call_function ( #60057 )
...
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/60057
This ensures that if a function was `wrap`'d before symbolic tracing + being passed into the transformer then it will still be wrapped.
Test Plan: Added test to `test_fx.py`
Reviewed By: jamesr66a
Differential Revision: D29151191
fbshipit-source-id: 93560be59505bdcfe8d4f013e21d4719788afd59
2021-06-16 17:25:55 -07:00
James Reed
7b73fdf597
[FX] Fix retracing wrapped functions ( #58061 )
...
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/58061
Test Plan: Imported from OSS
Reviewed By: yuhc
Differential Revision: D28358801
Pulled By: jamesr66a
fbshipit-source-id: c7c9a8a80e5bfe1eb1f6d2cf858ac7e57153a860
2021-05-17 19:50:16 -07:00
Sam Estep
75024e228c
Add lint for unqualified type: ignore ( #56290 )
...
Summary:
The other half of https://github.com/pytorch/pytorch/issues/56272 .
Pull Request resolved: https://github.com/pytorch/pytorch/pull/56290
Test Plan:
CI should pass on the tip of this PR, and we know that the lint works because the following CI runs (before this PR was finished) failed:
- https://github.com/pytorch/pytorch/runs/2384511062
- https://github.com/pytorch/pytorch/actions/runs/765036024
Reviewed By: seemethere
Differential Revision: D27867219
Pulled By: samestep
fbshipit-source-id: e648f07b6822867e70833e23ddafe7fb7eaca235
2021-04-21 08:07:23 -07:00
James Reed
a28c7db9f9
[FX] Garbage collect values in Interpreter ( #54726 )
...
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/54726
Test Plan: Imported from OSS
Reviewed By: ansley
Differential Revision: D27341449
Pulled By: jamesr66a
fbshipit-source-id: 9dc5f9675ed197dee4a31c8b0e6276248378f1ea
2021-03-25 20:35:32 -07:00
Shiyan Deng
238b0bbb68
Allow Transformer accept output result that is not Proxy ( #52473 )
...
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/52473
Use `map_aggregate` to create output for new graph so that it won't raise error when we have outputs that is not `Proxy`.
Test Plan: `test_transformer_multi_outputs` in `test_fx.py`
Reviewed By: jamesr66a
Differential Revision: D26502277
fbshipit-source-id: 404d9030a9b84db3f66f8505887a75717a28ad30
2021-02-23 19:28:37 -08:00
James Reed
256f93fb0f
[FX][EZ] Fix tuple type annotations ( #52010 )
...
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/52010
Test Plan: Imported from OSS
Reviewed By: ansley
Differential Revision: D26355481
Pulled By: jamesr66a
fbshipit-source-id: 27bbc5d8949beb68663f2e1e7963bec9afbef0cc
2021-02-09 20:32:30 -08:00
James Reed
d4e84b0c07
[FX] Fix leaf modules in Transformer ( #51998 )
...
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/51998
Test Plan: Imported from OSS
Reviewed By: Chillee
Differential Revision: D26352087
Pulled By: jamesr66a
fbshipit-source-id: ad8abc6507d4ea95fd3c99b226d1b40c3e9e64cf
2021-02-09 20:29:17 -08:00
James Reed
609f76f27a
[WIP][FX] Add Interpreter and Transformer ( #50420 )
...
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/50420
Test Plan: Imported from OSS
Reviewed By: zdevito
Differential Revision: D25880330
Pulled By: jamesr66a
fbshipit-source-id: 27d34888e36e39924821fed891d79f969237a104
2021-02-01 11:40:12 -08:00