Method serialize

exir/serde/export_serialize.py:1375–1426

Args: exported_program: Exported Program to serialize

(self, exported_program: ep.ExportedProgram)

            self.opset_version["aten"] = torch._C._get_max_operator_version()

    def serialize(self, exported_program: ep.ExportedProgram) -> _SerializedProgram:
        """
        Args:
            exported_program: Exported Program to serialize
        """
        exported_program._validate()

        gm_serializer = GraphModuleSerializer(
            exported_program.graph_signature, exported_program.module_call_graph
        )
        serialized_graph_module = gm_serializer.serialize(exported_program.graph_module)
        serialized_range_constraints = serialize_range_constraints(
            exported_program.range_constraints
        )

        # TODO: Directly serialize exported_program.constants once
        # CustomClassHolders get stored in the ExportedProgram rather than in
        # the graph
        constants = {}
        for n, c in gm_serializer.custom_objs.items():
            constants[n] = c
        for n, t in exported_program.constants.items():
            assert n not in constants
            constants[n] = t

        additional_kwargs = {}
        if hasattr(exported_program, "verifiers"):
            additional_kwargs["verifiers"] = [
                v.dialect for v in exported_program.verifiers
            ]
        elif hasattr(exported_program, "dialect"):
            additional_kwargs["dialect"] = exported_program.dialect
        serialized_ep = ExportedProgram(
            graph_module=serialized_graph_module,
            opset_version=self.opset_version,
            range_constraints=serialized_range_constraints,
            schema_version=SchemaVersion(
                major=SCHEMA_VERSION[0],
                minor=SCHEMA_VERSION[1],
            ),
            **additional_kwargs,
        )

        # Test canonical form is well defined.
        canonicalize(serialized_ep)

        return _SerializedProgram(
            serialized_ep,
            serialize_torch_artifact(exported_program.state_dict),
            serialize_torch_artifact(constants),
            serialize_torch_artifact(exported_program.example_inputs),
        )


class GraphModuleDeserializer: