pytorch/torch/_export
Jiashen Cao ca7ce2fca1 [ts-migration][1/N]: Add prim::Loop for constant number of iterations and condition (#131418)
#### Description
This PR adds prim::Loop support for the simplest case where the number of iteration is constant and the loop termination condition is also a constant.

[PR by stages](https://docs.google.com/document/d/1q6OprW3HBHbYPwEyE_DikBn-uzmhnN284Cmen_CnlhI/edit?usp=sharing)

#### Test Plan
Add reprod example.
* `pytest test/export/test_converter.py -s -k test_ts2ep_with_loop`
Pull Request resolved: https://github.com/pytorch/pytorch/pull/131418
Approved by: https://github.com/angelayi
2024-08-06 16:51:08 +00:00
..
db [export] Fix import in D60427208 (#132707) 2024-08-06 02:35:17 +00:00
pass_infra Flip default value for mypy disallow_untyped_defs [1/11] (#127838) 2024-06-08 18:16:33 +00:00
passes [export][reland] Convert autocast to HOO (#132677) 2024-08-05 22:34:52 +00:00
serde serde sympy functions (#132493) 2024-08-05 08:08:50 +00:00
__init__.py Revert "[BE] typing for decorators - fx/_compatibility (#131568)" 2024-07-28 03:43:39 +00:00
converter.py [ts-migration][1/N]: Add prim::Loop for constant number of iterations and condition (#131418) 2024-08-06 16:51:08 +00:00
error.py Fix global flake8 issues (#124771) 2024-04-26 15:35:53 +00:00
non_strict_utils.py [BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765) 2024-07-31 10:42:50 +00:00
pass_base.py Revert "[BE] typing for decorators - fx/_compatibility (#131568)" 2024-07-28 03:43:39 +00:00
tools.py [BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765) 2024-07-31 10:42:50 +00:00
utils.py [export][reland] Convert autocast to HOO (#132677) 2024-08-05 22:34:52 +00:00
verifier.py check unsupported sympy functions for runtime asserts (#132457) 2024-08-03 10:17:25 +00:00
wrappers.py Flip default value for mypy disallow_untyped_defs [2/11] (#127839) 2024-06-08 18:23:08 +00:00