2 Commits

Author	SHA1	Message	Date
Zhengxu Chen	fbfc530442	[export][ez] Fix forward D67044185 (#143193) ... Summary: Fixing forward D67044185 and T210459833 by adding the missing buld file. Test Plan: buck2 build --flagfile fbcode//mode/opt fbcode//admarket/training_data/augmentation/processors/tests:model_manager_test Differential Revision: D67200056 Pull Request resolved: https://github.com/pytorch/pytorch/pull/143193 Approved by: https://github.com/tugsbayasgalan	2024-12-13 16:06:42 +00:00
Zhengxu Chen	ee5bceaee6	[sigmoid] Write the new export schema format to archive without breaking compatibility. (#142511) ... Summary: This diff make it possible to migrate to PyTorch's OSS export schema from sigmoid. Basically, we add a new field called "methods" to ExportedProgram in Model definition, which contains the thrift schema generated based on schema.py from OSS. This way, we can keep writing the old fields while double write a new format in equivalent form. Since thrift doesn't support inlining type definitions, we do it manually here and it shouldn't break on-wire compatibility. As long as every sigmoid user is using sigmoid.frontend.serialization.serialize, we always guarantee to have the new format saved sa well. Eventually we will will use json deserialization from OSS so we will only keep this double writing for a couple of months. Eventually, we will migrate every serialization path to the OSS workflow. Test Plan: buck test mode/opt sigmoid/frontend:serialization_test buck test mode/opt sigmoid/frontend/test_gpu:serializer_test Differential Revision: D67044185 Pull Request resolved: https://github.com/pytorch/pytorch/pull/142511 Approved by: https://github.com/desertfire	2024-12-12 18:41:10 +00:00