mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
1a7da6e7e9
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
463 lines
7.9 KiB
YAML
463 lines
7.9 KiB
YAML
# @generated by update_schema.py
|
|
# checksum<<0335ca6e44a8a815ea638d538de0ad4f78a644af2689f6e93c0e8219117466e7>>
|
|
Argument:
|
|
kind: union
|
|
fields:
|
|
as_none:
|
|
type: Tuple[()]
|
|
as_tensor:
|
|
type: TensorArgument
|
|
as_tensors:
|
|
type: List[TensorArgument]
|
|
as_int:
|
|
type: int
|
|
as_ints:
|
|
type: List[int]
|
|
as_float:
|
|
type: float
|
|
as_floats:
|
|
type: List[float]
|
|
as_string:
|
|
type: str
|
|
as_strings:
|
|
type: List[str]
|
|
as_sym_int:
|
|
type: SymIntArgument
|
|
as_sym_ints:
|
|
type: List[SymIntArgument]
|
|
as_scalar_type:
|
|
type: ScalarType
|
|
as_memory_format:
|
|
type: MemoryFormat
|
|
as_layout:
|
|
type: Layout
|
|
as_device:
|
|
type: Device
|
|
as_bool:
|
|
type: bool
|
|
as_bools:
|
|
type: List[bool]
|
|
as_sym_bool:
|
|
type: SymBoolArgument
|
|
as_sym_bools:
|
|
type: List[SymBoolArgument]
|
|
as_graph:
|
|
type: GraphArgument
|
|
as_optional_tensors:
|
|
type: List[OptionalTensorArgument]
|
|
as_custom_obj:
|
|
type: CustomObjArgument
|
|
as_operator:
|
|
type: str
|
|
as_sym_float:
|
|
type: SymFloatArgument
|
|
as_sym_floats:
|
|
type: List[SymFloatArgument]
|
|
BufferMutationSpec:
|
|
kind: struct
|
|
fields:
|
|
arg:
|
|
type: TensorArgument
|
|
buffer_name:
|
|
type: str
|
|
ConstantValue:
|
|
kind: union
|
|
fields:
|
|
as_none:
|
|
type: Tuple[()]
|
|
as_int:
|
|
type: int
|
|
as_float:
|
|
type: float
|
|
as_string:
|
|
type: str
|
|
as_bool:
|
|
type: bool
|
|
CustomObjArgument:
|
|
kind: struct
|
|
fields:
|
|
name:
|
|
type: str
|
|
class_fqn:
|
|
type: str
|
|
Device:
|
|
kind: struct
|
|
fields:
|
|
type:
|
|
type: str
|
|
index:
|
|
type: Optional[int]
|
|
default: None
|
|
ExportedProgram:
|
|
kind: struct
|
|
fields:
|
|
graph_module:
|
|
type: GraphModule
|
|
opset_version:
|
|
type: Dict[str, int]
|
|
range_constraints:
|
|
type: Dict[str, RangeConstraint]
|
|
schema_version:
|
|
type: SchemaVersion
|
|
verifiers:
|
|
type: List[str]
|
|
default: '[]'
|
|
torch_version:
|
|
type: str
|
|
default: <=2.4
|
|
GradientToParameterSpec:
|
|
kind: struct
|
|
fields:
|
|
arg:
|
|
type: TensorArgument
|
|
parameter_name:
|
|
type: str
|
|
GradientToUserInputSpec:
|
|
kind: struct
|
|
fields:
|
|
arg:
|
|
type: TensorArgument
|
|
user_input_name:
|
|
type: str
|
|
Graph:
|
|
kind: struct
|
|
fields:
|
|
inputs:
|
|
type: List[Argument]
|
|
outputs:
|
|
type: List[Argument]
|
|
nodes:
|
|
type: List[Node]
|
|
tensor_values:
|
|
type: Dict[str, TensorMeta]
|
|
sym_int_values:
|
|
type: Dict[str, SymInt]
|
|
sym_bool_values:
|
|
type: Dict[str, SymBool]
|
|
is_single_tensor_return:
|
|
type: bool
|
|
default: 'False'
|
|
custom_obj_values:
|
|
type: Dict[str, CustomObjArgument]
|
|
default: '{}'
|
|
sym_float_values:
|
|
type: Dict[str, SymFloat]
|
|
default: '{}'
|
|
GraphArgument:
|
|
kind: struct
|
|
fields:
|
|
name:
|
|
type: str
|
|
graph:
|
|
type: Graph
|
|
GraphModule:
|
|
kind: struct
|
|
fields:
|
|
graph:
|
|
type: Graph
|
|
signature:
|
|
type: GraphSignature
|
|
module_call_graph:
|
|
type: List[ModuleCallEntry]
|
|
metadata:
|
|
type: Dict[str, str]
|
|
default: '{}'
|
|
GraphSignature:
|
|
kind: struct
|
|
fields:
|
|
input_specs:
|
|
type: List[InputSpec]
|
|
output_specs:
|
|
type: List[OutputSpec]
|
|
InputSpec:
|
|
kind: union
|
|
fields:
|
|
user_input:
|
|
type: UserInputSpec
|
|
parameter:
|
|
type: InputToParameterSpec
|
|
buffer:
|
|
type: InputToBufferSpec
|
|
tensor_constant:
|
|
type: InputToTensorConstantSpec
|
|
custom_obj:
|
|
type: InputToCustomObjSpec
|
|
token:
|
|
type: InputTokenSpec
|
|
constant_input:
|
|
type: InputToConstantInputSpec
|
|
InputToBufferSpec:
|
|
kind: struct
|
|
fields:
|
|
arg:
|
|
type: TensorArgument
|
|
buffer_name:
|
|
type: str
|
|
persistent:
|
|
type: bool
|
|
InputToConstantInputSpec:
|
|
kind: struct
|
|
fields:
|
|
name:
|
|
type: str
|
|
value:
|
|
type: ConstantValue
|
|
InputToCustomObjSpec:
|
|
kind: struct
|
|
fields:
|
|
arg:
|
|
type: CustomObjArgument
|
|
custom_obj_name:
|
|
type: str
|
|
InputToParameterSpec:
|
|
kind: struct
|
|
fields:
|
|
arg:
|
|
type: TensorArgument
|
|
parameter_name:
|
|
type: str
|
|
InputToTensorConstantSpec:
|
|
kind: struct
|
|
fields:
|
|
arg:
|
|
type: TensorArgument
|
|
tensor_constant_name:
|
|
type: str
|
|
InputTokenSpec:
|
|
kind: struct
|
|
fields:
|
|
arg:
|
|
type: TokenArgument
|
|
Layout:
|
|
kind: enum
|
|
fields:
|
|
Unknown: 0
|
|
SparseCoo: 1
|
|
SparseCsr: 2
|
|
SparseCsc: 3
|
|
SparseBsr: 4
|
|
SparseBsc: 5
|
|
_mkldnn: 6
|
|
Strided: 7
|
|
LossOutputSpec:
|
|
kind: struct
|
|
fields:
|
|
arg:
|
|
type: TensorArgument
|
|
MemoryFormat:
|
|
kind: enum
|
|
fields:
|
|
Unknown: 0
|
|
ContiguousFormat: 1
|
|
ChannelsLast: 2
|
|
ChannelsLast3d: 3
|
|
PreserveFormat: 4
|
|
ModuleCallEntry:
|
|
kind: struct
|
|
fields:
|
|
fqn:
|
|
type: str
|
|
signature:
|
|
type: Optional[ModuleCallSignature]
|
|
default: None
|
|
ModuleCallSignature:
|
|
kind: struct
|
|
fields:
|
|
inputs:
|
|
type: List[Argument]
|
|
outputs:
|
|
type: List[Argument]
|
|
in_spec:
|
|
type: str
|
|
out_spec:
|
|
type: str
|
|
forward_arg_names:
|
|
type: Optional[List[str]]
|
|
default: None
|
|
NamedArgument:
|
|
kind: struct
|
|
fields:
|
|
name:
|
|
type: str
|
|
arg:
|
|
type: Argument
|
|
Node:
|
|
kind: struct
|
|
fields:
|
|
target:
|
|
type: str
|
|
inputs:
|
|
type: List[NamedArgument]
|
|
outputs:
|
|
type: List[Argument]
|
|
metadata:
|
|
type: Dict[str, str]
|
|
OptionalTensorArgument:
|
|
kind: union
|
|
fields:
|
|
as_tensor:
|
|
type: TensorArgument
|
|
as_none:
|
|
type: Tuple[()]
|
|
OutputSpec:
|
|
kind: union
|
|
fields:
|
|
user_output:
|
|
type: UserOutputSpec
|
|
loss_output:
|
|
type: LossOutputSpec
|
|
buffer_mutation:
|
|
type: BufferMutationSpec
|
|
gradient_to_parameter:
|
|
type: GradientToParameterSpec
|
|
gradient_to_user_input:
|
|
type: GradientToUserInputSpec
|
|
user_input_mutation:
|
|
type: UserInputMutationSpec
|
|
token:
|
|
type: OutputTokenSpec
|
|
OutputTokenSpec:
|
|
kind: struct
|
|
fields:
|
|
arg:
|
|
type: TokenArgument
|
|
RangeConstraint:
|
|
kind: struct
|
|
fields:
|
|
min_val:
|
|
type: Optional[int]
|
|
max_val:
|
|
type: Optional[int]
|
|
ScalarType:
|
|
kind: enum
|
|
fields:
|
|
UNKNOWN: 0
|
|
BYTE: 1
|
|
CHAR: 2
|
|
SHORT: 3
|
|
INT: 4
|
|
LONG: 5
|
|
HALF: 6
|
|
FLOAT: 7
|
|
DOUBLE: 8
|
|
COMPLEXHALF: 9
|
|
COMPLEXFLOAT: 10
|
|
COMPLEXDOUBLE: 11
|
|
BOOL: 12
|
|
BFLOAT16: 13
|
|
UINT16: 28
|
|
SchemaVersion:
|
|
kind: struct
|
|
fields:
|
|
major:
|
|
type: int
|
|
minor:
|
|
type: int
|
|
SymBool:
|
|
kind: union
|
|
fields:
|
|
as_expr:
|
|
type: SymExpr
|
|
as_bool:
|
|
type: bool
|
|
SymBoolArgument:
|
|
kind: union
|
|
fields:
|
|
as_name:
|
|
type: str
|
|
as_bool:
|
|
type: bool
|
|
SymExpr:
|
|
kind: struct
|
|
fields:
|
|
expr_str:
|
|
type: str
|
|
hint:
|
|
type: Optional[SymExprHint]
|
|
default: None
|
|
SymExprHint:
|
|
kind: union
|
|
fields:
|
|
as_int:
|
|
type: int
|
|
as_bool:
|
|
type: bool
|
|
as_float:
|
|
type: float
|
|
SymFloat:
|
|
kind: union
|
|
fields:
|
|
as_expr:
|
|
type: SymExpr
|
|
as_float:
|
|
type: float
|
|
SymFloatArgument:
|
|
kind: union
|
|
fields:
|
|
as_name:
|
|
type: str
|
|
as_float:
|
|
type: float
|
|
SymInt:
|
|
kind: union
|
|
fields:
|
|
as_expr:
|
|
type: SymExpr
|
|
as_int:
|
|
type: int
|
|
SymIntArgument:
|
|
kind: union
|
|
fields:
|
|
as_name:
|
|
type: str
|
|
as_int:
|
|
type: int
|
|
TensorArgument:
|
|
kind: struct
|
|
fields:
|
|
name:
|
|
type: str
|
|
TensorMeta:
|
|
kind: struct
|
|
fields:
|
|
dtype:
|
|
type: ScalarType
|
|
sizes:
|
|
type: List[SymInt]
|
|
requires_grad:
|
|
type: bool
|
|
device:
|
|
type: Device
|
|
strides:
|
|
type: List[SymInt]
|
|
storage_offset:
|
|
type: SymInt
|
|
layout:
|
|
type: Layout
|
|
TokenArgument:
|
|
kind: struct
|
|
fields:
|
|
name:
|
|
type: str
|
|
UserInputMutationSpec:
|
|
kind: struct
|
|
fields:
|
|
arg:
|
|
type: TensorArgument
|
|
user_input_name:
|
|
type: str
|
|
UserInputSpec:
|
|
kind: struct
|
|
fields:
|
|
arg:
|
|
type: Argument
|
|
UserOutputSpec:
|
|
kind: struct
|
|
fields:
|
|
arg:
|
|
type: Argument
|
|
SCHEMA_VERSION:
|
|
- 8
|
|
- 2
|
|
TREESPEC_VERSION: 1
|