2 Commits

Author	SHA1	Message	Date
Angela Yi	e8ab8c877d	[exir] Add lift constant tensors passes after aten_to_edge (#109382) ... Summary: X-link: https://github.com/pytorch/executorch/pull/359 When exporting using enable_aot (through the torch.export path), we want to lift all constant tensors as buffers to the exported program. The ScalarToTensor pass in EXIR's aten_to_edge passes will create some constant tensors in the graph, so we will need to run a lift_constant_tensors pass afterwards. Note that this only needs to be applied when exporting using the torch.export path because in the original path, nothing is lifted. Test Plan: CI Differential Revision: D49207492 Pull Request resolved: https://github.com/pytorch/pytorch/pull/109382 Approved by: https://github.com/cccclai	2023-09-19 01:34:58 +00:00
Angela Yi	025d1a18ab	[export] Separate out exported_program.py (#109147) ... Test Plan: CI Differential Revision: D49205011 Pull Request resolved: https://github.com/pytorch/pytorch/pull/109147 Approved by: https://github.com/zhxchen17	2023-09-14 01:14:46 +00:00