suo
4057d005ff
Initial torchbind support in PT2 ( #117697 )
...
This PR adds the bare minimum functionality to get torchbind working in an e2e testable way on PT2.
It implements:
* ProxyTensor support
* Simple torch.export support (proxytensor-only path, e.g. non-strict).
* add some tests exercising the path.
Because all this is not fully baked, I hide the functionality behind a feature flag (`enable_torchbind_tracing()`) so it does not affect regular users for now.
Still on the agenda:
* Dynamo support
* Actual FakeMode support
* Mutability support
Hoping to get this first bit in as a standalone, as it will unblock some more extensive experimentation/testing going on internally.
Differential Revision: [D51825372](https://our.internmc.facebook.com/intern/diff/D51825372/ )
Pull Request resolved: https://github.com/pytorch/pytorch/pull/117697
Approved by: https://github.com/SherlockNoMad
2024-01-19 06:28:20 +00:00
Angela Yi
92d718aed1
[export] Add lifted constant obj to input ( #116985 )
...
Test Plan: wip
Differential Revision: D52556070
Pull Request resolved: https://github.com/pytorch/pytorch/pull/116985
Approved by: https://github.com/suo
2024-01-18 22:10:53 +00:00
Zhengxu Chen
e0d2a24967
Reland "[export] Support user input mutation. [1/2]" ( #114496 ) ( #114596 )
...
Summary:
Serialization not implemented yet. Will do in the next diff.
Resolving Github issues:
https://github.com/pytorch/pytorch/issues/112429
https://github.com/pytorch/pytorch/issues/114142
Test Plan:
onnx doc test
```
python -m xdoctest /opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/onnx/_internal/exporter.py ONNXProgram.model_signature:0
```
Differential Revision: D51588558
Pull Request resolved: https://github.com/pytorch/pytorch/pull/114596
Approved by: https://github.com/angelayi
2023-11-27 20:19:04 +00:00
PyTorch MergeBot
fa1ccc34c4
Revert "[export] Support user input mutation. [1/2] ( #114496 )"
...
This reverts commit b62c0d96bchttps://github.com/pytorch/pytorch/pull/114496 on behalf of https://github.com/facebook-github-bot due to Diff reverted internally ([comment](https://github.com/pytorch/pytorch/pull/114496#issuecomment-1827289635 ))
2023-11-27 07:52:21 +00:00
Zhengxu Chen
b62c0d96bc
[export] Support user input mutation. [1/2] ( #114496 )
...
Summary:
Serialization not implemented yet. Will do in the next diff.
Resolving Github issues:
https://github.com/pytorch/pytorch/issues/112429
https://github.com/pytorch/pytorch/issues/114142
Test Plan:
buck2 run mode/opt caffe2/test:test_export -- -r test_export_
input_mutation
Differential Revision: D51556962
Pull Request resolved: https://github.com/pytorch/pytorch/pull/114496
Approved by: https://github.com/tugsbayasgalan
2023-11-27 04:53:38 +00:00
Angela Yi
50101d59ba
[export][retry] Move lifted tensors out of state_dict ( #113689 )
...
Test Plan: CI
Differential Revision: D51321532
Pull Request resolved: https://github.com/pytorch/pytorch/pull/113689
Approved by: https://github.com/zhxchen17
2023-11-15 09:24:49 +00:00
Aaron Gokaslan
8219bf051b
[BE]: Apply RUF015 to torch folder ( #113025 )
...
Removes unnecessary allocations of iterators. There is a small chance this may have side effects as the entire iterator is no longer consumed, but this is a way more efficient method for retrieving the first element.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/113025
Approved by: https://github.com/ezyang , https://github.com/malfet
2023-11-07 00:48:15 +00:00
Zhengxu Chen
50767a075a
[export] Clean up verifier [1/n]. ( #112505 )
...
Summary: Some adjustments to verifier so that it's easier to use it correctly. We will enable verifier later, so the current diff is no-op.
Test Plan: CI
Differential Revision: D50839295
Pull Request resolved: https://github.com/pytorch/pytorch/pull/112505
Approved by: https://github.com/tugsbayasgalan , https://github.com/angelayi
2023-11-02 19:36:06 +00:00
angelayi
131e0f1b75
[export] Separate out graph signature ( #112412 )
...
Differential Revision: [D50800524](https://our.internmc.facebook.com/intern/diff/D50800524 )
Pull Request resolved: https://github.com/pytorch/pytorch/pull/112412
Approved by: https://github.com/zhxchen17
2023-11-02 00:18:28 +00:00
