Zhengxu Chen 1a7da6e7e9 [export] Add test to enforce consistency between synced thrift and generated thrift from schema.py (#141989) ...

Summary:
In this diff we implement a way to ensure the internal thrift schema from cfgr (configerator/structs/caffe2/torch/export/schema.thrift) and the schema in OSS (torch/_export/serde/schema.thrift) are in sync, by adding a unittest to reflect on the type names and fields from each schema and compare them field by field.

When we detect new fields/types from torch/_export/serde/schema.thrift, there'll be a test failure on the trunk and the error message hints people to add the missing field/type to the thrift schema from cfgr, so that they are always in sync in practice.

Test Plan: buck test mode/opt caffe2/test:test_export -- -r test_thrift_schema_in_sync

Differential Revision: D66716834

Pull Request resolved: https://github.com/pytorch/pytorch/pull/141989
Approved by: https://github.com/yiming0416

2024-12-06 18:42:20 +00:00

..

update_schema.py

[export] Add test to enforce consistency between synced thrift and generated thrift from schema.py (#141989)

2024-12-06 18:42:20 +00:00