def METHOD_NAME(self, irmodule: tvm.IRModule) -> tvm.IRModule:
    """
    Return an updated IRModule by replacing the old function with the mutated root function.
    Parameters
    ----------
    irmodule : tvm.IRModule
        The base IRModule to update.
    Returns
    -------
    tvm.IRModule
        The updated IRModule.
    """
    ret = _ffi_api.dfb_rewrite_mutate_irmodule(self, irmodule)  # type: ignore
    if hasattr(irmodule, "__name__"):
        ret.__name__ = irmodule.__name__
    return ret