Method add_exported_program

devtools/etrecord/_etrecord.py:401–434

Add exported program to the ETRecord after it has been created. This method allows users to add an exported program they want to record to an existing ETRecord instance. The exported program will be included when the ETRecord is saved. Args: exp

(
        self,
        exported_program: Optional[Union[ExportedProgram, Dict[str, ExportedProgram]]],
    )

        self._representative_inputs = representative_inputs

    def add_exported_program(
        self,
        exported_program: Optional[Union[ExportedProgram, Dict[str, ExportedProgram]]],
    ) -> None:
        """
        Add exported program to the ETRecord after it has been created.

        This method allows users to add an exported program they want to record
        to an existing ETRecord instance. The exported program will be included
        when the ETRecord is saved.

        Args:
            exported_program: The exported program for this model returned by the call to
                `torch.export()` or a dictionary with method names as keys and exported programs as values.
                Can be None, in which case no exported program data will be added.

        Raises:
            RuntimeError: If exported program already exists in the ETRecord.
        """
        # Check if exported program already exists
        if self.exported_program is not None or self.export_graph_id is not None:
            raise RuntimeError(
                "Exported program already exists in the ETRecord. "
                "Cannot add exported program when it already exists."
            )

        # Process exported program and extract data
        processed_exported_program, export_graph_id = _process_exported_program(
            exported_program
        )

        # Set the extracted data
        self.exported_program = processed_exported_program
        self.export_graph_id = export_graph_id

    def add_edge_dialect_program(
        self,