pytorch/test/export
Boyuan Feng 2cef2fc2b4 [ts migration] support aten::dim, aten::len, aten::__getitem__ (#127593)
- Add support for aten::dim, aten::len, aten::__getitem__ for torchscript to export converter.
- Add unit tests
Co-authored-by: cyy <cyyever@outlook.com>
Co-authored-by: Menglu Yu <mengluy@meta.com>
Co-authored-by: Animesh Jain <anijain@umich.edu>
Co-authored-by: Simon Fan <xmfan@meta.com>
Co-authored-by: Zain Rizvi <ZainR@meta.com>
Co-authored-by: Tugsbayasgalan (Tugsuu) Manlaibaatar <tmanlaibaatar@meta.com>
Co-authored-by: titaiwangms <titaiwang@microsoft.com>
Co-authored-by: Yueming Hao <yhao@meta.com>
Co-authored-by: IvanKobzarev <ivan.kobzarev@gmail.com>
Co-authored-by: PyTorch MergeBot <pytorchmergebot@users.noreply.github.com>
Co-authored-by: Edward Z. Yang <ezyang@meta.com>
Co-authored-by: Bin Bao <binbao@meta.com>
Co-authored-by: Feny Patel <fenypatel@meta.com>
Co-authored-by: Mikayla Gawarecki <mikaylagawarecki@gmail.com>
Co-authored-by: xinan.lin <xinan.lin@intel.com>
Co-authored-by: Zain Huda <zainhuda@meta.com>
Co-authored-by: Chien-Chin Huang <chienchin@fb.com>
Co-authored-by: Wei Wang <weiwan@nvidia.com>
Co-authored-by: Jason Ansel <jansel@meta.com>
Co-authored-by: Aaron Gokaslan <aaronGokaslan@gmail.com>
Co-authored-by: Iris Z <31293777+wz337@users.noreply.github.com>
Co-authored-by: Wang, Eikan <eikan.wang@intel.com>
Co-authored-by: angelayi <yiangela7@gmail.com>
Co-authored-by: Svetlana Karslioglu <svekars@meta.com>
Co-authored-by: Yanbo Liang <ybliang8@gmail.com>
Co-authored-by: Catherine Lee <csl@fb.com>
Co-authored-by: Kwanghoon An <kwanghoon@meta.com>
Co-authored-by: Brian Hirsh <hirsheybar@fb.com>
Co-authored-by: Robert Mast <rmast@live.nl>
Co-authored-by: drisspg <drisspguessous@gmail.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/127593
Approved by: https://github.com/SherlockNoMad, https://github.com/malfet
2024-06-02 00:36:33 +00:00
..
__init__.py
opinfo_schema.py [export] add SchemaCheckMode testing for pre-dispatch export, OpInfo (#125481) 2024-05-14 21:07:21 +00:00
test_converter.py [ts migration] support aten::dim, aten::len, aten::__getitem__ (#127593) 2024-06-02 00:36:33 +00:00
test_db.py
test_experimental.py [Fix]: populate input parameter name when convert TorchScript to ExportedProgram (#126787) 2024-05-28 17:33:44 +00:00
test_export.py Add deferred_runtime_assertion pass after run_decompositions (#127305) 2024-05-31 05:45:28 +00:00
test_export_nonstrict.py
test_functionalized_assertions.py
test_hop.py
test_lift_unlift.py [export] handle constant aliasing for export (#125509) 2024-05-10 00:14:37 +00:00
test_pass_infra.py [5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126) 2024-05-27 14:49:57 +00:00
test_passes.py [5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126) 2024-05-27 14:49:57 +00:00
test_retraceability.py [Export] Add runtime assert to non-strict export (#123681) 2024-04-18 16:13:27 +00:00
test_schema.py
test_serdes.py [export] Delete predispatch tests (#126459) 2024-05-17 00:48:32 +00:00
test_serialize.py Add a registry for GraphModuleSerializer (#126550) 2024-05-29 03:12:48 +00:00
test_sparse.py [traced-graph][sparse] propagate sparsity metadata into traced graph (#117907) 2024-05-23 22:46:46 +00:00
test_tools.py Tool for scouting exportability in one shot (#126471) 2024-05-18 00:10:46 +00:00
test_torchbind.py [export] handle constant aliasing for export (#125509) 2024-05-10 00:14:37 +00:00
test_tree_utils.py
test_unflatten.py [5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126) 2024-05-27 14:49:57 +00:00
test_verifier.py [5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126) 2024-05-27 14:49:57 +00:00
testing.py