pytorch/torch/_export
Tugsbayasgalan Manlaibaatar 0a6c40faba Fix constant returning (#137993)
When the constants are used twice in the exported graph (second one is returned as output), the lifting constant pass doesn't account for the second one being the output. THis PR fixes that.

Differential Revision: [D64406108](https://our.internmc.facebook.com/intern/diff/D64406108/)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/137993
Approved by: https://github.com/avikchaudhuri
2024-10-16 16:42:09 +00:00
..
db fix requirement for error classification (#133122) 2024-08-10 04:59:09 +00:00
pass_infra Flip default value for mypy disallow_untyped_defs [1/11] (#127838) 2024-06-08 18:16:33 +00:00
passes Fix constant returning (#137993) 2024-10-16 16:42:09 +00:00
serde Add original forward names to schema so that prettify pass works (#136887) 2024-10-08 04:21:02 +00:00
__init__.py Make PT2E work with both IR simultaneously (#135769) 2024-10-02 21:05:22 +00:00
converter.py [ts_converter] Fix prim::If buffer names (#136648) 2024-10-02 00:07:47 +00:00
error.py Fix global flake8 issues (#124771) 2024-04-26 15:35:53 +00:00
non_strict_utils.py [export] clean up dynamic markers from tensors (#137230) 2024-10-04 06:50:45 +00:00
pass_base.py [BE] typing for decorators - fx/_compatibility (part 1) (#134202) 2024-08-22 17:07:33 +00:00
tools.py [report_exportability] Avoid re-exporting duplicated modules (#133930) 2024-08-20 22:11:57 +00:00
utils.py Fix param and buffer mapping for state_dict when there are state_dict hooks (#137609) 2024-10-11 01:33:50 +00:00
verifier.py Allow autocast in training ir export (#137287) 2024-10-04 17:38:51 +00:00
wrappers.py [hop] require hops to override __call__. (#134352) 2024-08-28 19:56:40 +00:00