PyTorch MergeBot
80a9625d9f
Revert "non-strict export with dynamic shapes ( #115862 )"
...
This reverts commit 1bb0d0fc1fhttps://github.com/pytorch/pytorch/pull/115862 on behalf of https://github.com/atalman due to OSSCI oncall, failing trunk / macos-12-py3-arm64 / test ([comment](https://github.com/pytorch/pytorch/pull/115862#issuecomment-1858482486 ))
2023-12-15 21:04:12 +00:00
Avik Chaudhuri
1bb0d0fc1f
non-strict export with dynamic shapes ( #115862 )
...
Differential Revision: [D52175048](https://our.internmc.facebook.com/intern/diff/D52175048/ )
Pull Request resolved: https://github.com/pytorch/pytorch/pull/115862
Approved by: https://github.com/zhxchen17
2023-12-15 20:11:30 +00:00
PyTorch MergeBot
1b506e7469
Revert "non-strict export with dynamic shapes ( #115862 )"
...
This reverts commit f54bb1ed56https://github.com/pytorch/pytorch/pull/115862 on behalf of https://github.com/atalman due to OSSCI oncall, failing trunk / macos-12-py3-arm64 / test ([comment](https://github.com/pytorch/pytorch/pull/115862#issuecomment-1858197497 ))
2023-12-15 17:03:42 +00:00
Avik Chaudhuri
f54bb1ed56
non-strict export with dynamic shapes ( #115862 )
...
Differential Revision: [D52175048](https://our.internmc.facebook.com/intern/diff/D52175048/ )
Pull Request resolved: https://github.com/pytorch/pytorch/pull/115862
Approved by: https://github.com/zhxchen17
2023-12-15 16:38:45 +00:00
angelayi
dd42201cb8
[export] Preserve FQN in export_to_torch_ir ( #115462 )
...
AOTInductor currently relies of export_to_torch_ir to generate a graph, and passes it to inductor to generate the .so. They would like the FQN to be consistent so that they can easily find/update the weights in the .so.
Note that since export flattens all modules in to a single computational graph, we will change the FQNs in the original module by replacing all periods with underscores. For example, `foo.child1param`, which points to a submodule named `foo`'s parameter named `child1param`, will be renamed to `foo_child1param` since we no longer have the submodule `foo`. This is done just by doing `name.replace(".", "_")`.
Outputted AOTInductor c++ code: https://www.internalfb.com/phabricator/paste/view/P900120950?lines=377-355%2C354
Pull Request resolved: https://github.com/pytorch/pytorch/pull/115462
Approved by: https://github.com/tugsbayasgalan
2023-12-13 04:58:47 +00:00
zhxchen17
f78f23d753
[export] Turn off output value from sources for export. ( #115442 )
...
Fixes #ISSUE_NUMBER
Pull Request resolved: https://github.com/pytorch/pytorch/pull/115442
Approved by: https://github.com/tugsbayasgalan
2023-12-12 22:41:23 +00:00
angelayi
36199747f3
[export][reland][refactor][2/n] Move tracing logic ( #115557 )
...
Reland of https://github.com/pytorch/pytorch/pull/114768
Pull Request resolved: https://github.com/pytorch/pytorch/pull/115557
Approved by: https://github.com/zhxchen17
ghstack dependencies: #115556
2023-12-12 05:37:07 +00:00
atalman
24a463c46c
Revert "[export][refactor][2/n] Move tracing logic ( #114768 )" ( #115503 )
...
Github first oncall.
This reverts commit 0ab57ee7eahttps://github.com/pytorch/pytorch/pull/115503
Approved by: https://github.com/angelayi , https://github.com/kit1980
2023-12-10 19:30:15 +00:00
PyTorch MergeBot
3e47e3f441
Revert "[export] Fix graph output mismatch issue with constant outputs. ( #115280 )"
...
This reverts commit 622688fab9https://github.com/pytorch/pytorch/pull/115280 on behalf of https://github.com/atalman due to ghfirst issue when importing, will reland this PR ([comment](https://github.com/pytorch/pytorch/pull/115280#issuecomment-1847903624 ))
2023-12-08 22:10:03 +00:00
zhxchen17
622688fab9
[export] Fix graph output mismatch issue with constant outputs. ( #115280 )
...
Fixes #ISSUE_NUMBER
Pull Request resolved: https://github.com/pytorch/pytorch/pull/115280
Approved by: https://github.com/tugsbayasgalan
2023-12-07 06:11:08 +00:00
angelayi
0ab57ee7ea
[export][refactor][2/n] Move tracing logic ( #114768 )
...
2/n of refactoring export code:
* Moved tracing logic in torch/_export/init.py to torch/export/_tracer.py
Differential Revision: [D51823961](https://our.internmc.facebook.com/intern/diff/D51823961 )
Pull Request resolved: https://github.com/pytorch/pytorch/pull/114768
Approved by: https://github.com/ydwu4
ghstack dependencies: #114764
2023-12-06 16:46:47 +00:00
