OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
381d0cb239
pytorch/test/export
History
Tugsbayasgalan Manlaibaatar 6b1b95ad2a Support subclass constructor capturing in export (#147014) 
Notable TODOs:
1. Need to implement AutogradHOP to get rid of subclasses before serializing
2. Need to implement mechanism to figure out what subclasses will be used in export when they are not expressed in the inputs

Differential Revision: [D69640673](https://our.internmc.facebook.com/intern/diff/D69640673)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/147014
Approved by: https://github.com/bdhirsh
2025-03-16 18:19:19 +00:00
..
__init__.py
opinfo_schema.py Fix unused Python variables in test/[e-z]* (#136964) 2024-12-18 23:02:30 +00:00
random_dag.py [BE] fix ruff rule E226: add missing whitespace around operator in f-strings (#144415) 2025-01-08 21:55:00 +00:00
test_converter.py [export] Fix tensor_constant and buffer naming conflicts in TS converter (#148803) 2025-03-14 00:38:12 +00:00
test_cpp_serdes.py [export] Implement cpp deserializer. (#136398) 2024-11-14 16:34:59 +00:00
test_db.py Set non-strict export as default mode (#148790) 2025-03-12 21:10:58 +00:00
test_draft_export.py Consolidate torchbind fake class registration (#149063) 2025-03-13 06:57:13 +00:00
test_experimental.py [export] handle buffer/input mutations for joint-graph (#144806) 2025-01-16 00:22:16 +00:00
test_export_legacy.py EZ fix to make sure local pytest run succeeds in export (#144764) 2025-01-15 00:43:40 +00:00
test_export_strict.py Set non-strict export as default mode (#148790) 2025-03-12 21:10:58 +00:00
test_export_training_ir_to_run_decomp.py Remove presere ops (#138371) 2024-10-25 19:13:55 +00:00
test_export.py Support subclass constructor capturing in export (#147014) 2025-03-16 18:19:19 +00:00
test_functionalized_assertions.py fix incorrect literal strings / accidental tuples (#146037) 2025-02-03 15:08:11 +00:00
test_hop.py Remove unused import in tests (#146266) 2025-02-03 03:40:18 +00:00
test_lift_unlift.py Consistently use load_torchbind_test_lib in tests (#148082) 2025-03-03 19:37:28 +00:00
test_pass_infra.py [Codemod][AddExplicitStrictExportArg] caffe2/test (#143688) 2024-12-27 07:58:44 +00:00
test_passes.py patch for block-wise quantization + pt2e (#146946) 2025-02-18 01:15:26 +00:00
test_retraceability.py Set non-strict export as default mode (#148790) 2025-03-12 21:10:58 +00:00
test_schema.py Tweak schema_check to handle annotated builtin types (#145154) 2025-01-19 18:48:35 +00:00
test_serdes.py Set non-strict export as default mode (#148790) 2025-03-12 21:10:58 +00:00
test_serialize.py Support basic TorchBind in aot_compile and aoti_compile_and_package (#148506) 2025-03-11 20:55:18 +00:00
test_sparse.py [Codemod][AddExplicitStrictExportArg] caffe2/test (#143688) 2024-12-27 07:58:44 +00:00
test_swap.py Remove unused import in tests (#146266) 2025-02-03 03:40:18 +00:00
test_tools.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_torchbind.py Add back fake class registration to test_torchbind (#149137) 2025-03-13 21:26:37 +00:00
test_tree_utils.py fix incorrect literal strings / accidental tuples (#146037) 2025-02-03 15:08:11 +00:00
test_unflatten_training_ir.py fix unflatten training ir test suffix (#138840) 2024-10-24 23:42:54 +00:00
test_unflatten.py [export] Include metadata in FlatArgsAdapter (#146107) 2025-02-05 00:29:58 +00:00
test_verifier.py Move test_verifier to training IR (#138249) 2024-10-18 07:36:29 +00:00
testing.py Set non-strict export as default mode (#148790) 2025-03-12 21:10:58 +00:00