Method update_func

python/tvm/relax/block_builder.py:738–749

Add a Relax function or a TIR PrimFunc to the IRModule being built. Parameters ---------- gv : GlobalVar The global var referring the function to be updated. updated_func : BaseFunc The updated function.

(self, gv: GlobalVar, updated_func: BaseFunc)

        return _ffi_api.BlockBuilderAddFunction(self, func, func_name)  # type: ignore

    def update_func(self, gv: GlobalVar, updated_func: BaseFunc) -> None:
        """Add a Relax function or a TIR PrimFunc to the IRModule being built.

        Parameters
        ----------
        gv : GlobalVar
            The global var referring the function to be updated.

        updated_func : BaseFunc
            The updated function.
        """
        return _ffi_api.BlockBuilderUpdateFunction(self, gv, updated_func)  # type: ignore

    def current_block_is_dataflow(self) -> bool:
        """Check if the block being built is DataflowBlock or not.