Function _create_python_packed_func

python/tvm/script/parser/core/entry.py:151–177

Create a PackedFunc wrapper for a Python function. This function creates a PackedFunc that can be called from TVM runtime and will execute the original Python function. Parameters ---------- pyfunc : Callable The Python function to wrap. Returns ------- Pac

(pyfunc)

def _create_python_packed_func(pyfunc):
    """Create a PackedFunc wrapper for a Python function.

    This function creates a PackedFunc that can be called from TVM runtime
    and will execute the original Python function.

    Parameters
    ----------
    pyfunc : Callable
        The Python function to wrap.

    Returns
    -------
    PackedFunc
        A PackedFunc that wraps the Python function.
    """

    def packed_func_wrapper(*args, **kwargs):
        """Wrapper function that calls the original Python function."""
        try:
            result = pyfunc(*args, **kwargs)
            return result
        except Exception as error:
            print(f"Error calling Python function {pyfunc.__name__}: {error}")
            raise

    return packed_func_wrapper


def _attach_pyfuncs_to_irmodule(irmodule, all_pyfuncs):