Method get_program

backends/test/program_builder.py:109–132

(self)

        ]

    def get_program(self) -> ExportedProgram:
        gm = self.get_graph_module()
        graph_signature = ExportGraphSignature(self.input_specs, self.output_specs)
        in_spec = pytree.tree_flatten((tuple(graph_signature.user_inputs), {}))[1]
        out_spec = pytree.tree_flatten(graph_signature.user_outputs)[1]
        return ExportedProgram(
            root=gm,
            graph=gm.graph,
            graph_signature=graph_signature,
            # pyre-ignore[6]: Incompatible parameter type.
            constants=self.constants,
            state_dict=self.state_dict,
            range_constraints={},
            module_call_graph=[
                ModuleCallEntry(
                    "",
                    ModuleCallSignature(
                        inputs=[], outputs=[], in_spec=in_spec, out_spec=out_spec
                    ),
                )
            ],
            # pyre-ignore[6]: Incompatible parameter type.
            verifiers=self.get_verifiers(),
        )

    def get_edge_program(self) -> EdgeProgramManager:
        return EdgeProgramManager(