Function call_packed

python/tvm/relax/script/builder/ir.py:407–461

Create a relax Call, which calls a packed function. Parameters ---------- func: str The name of extern function. *args : Expr The arguments. sinfo_args: Optional[Union[StructInfo, List[StructInfo]]] The list of structure info arguments. kwargs: Expr

(
    func: py_str,
    *args: Expr,
    sinfo_args: StructInfo | list[StructInfo] | None = None,
    **kwargs: Any,
)

def call_packed(
    func: py_str,
    *args: Expr,
    sinfo_args: StructInfo | list[StructInfo] | None = None,
    **kwargs: Any,
) -> Call:
    """Create a relax Call, which calls a packed function.
    Parameters
    ----------
    func: str
        The name of extern function.
    *args : Expr
        The arguments.
    sinfo_args: Optional[Union[StructInfo, List[StructInfo]]]
        The list of structure info arguments.
    kwargs: Expr
        The keyword arguments.

    Returns
    -------
    call: Call
        The created Relax Call
    """
    op = ExternFunc(func)
    args = py_tuple(convert_to_expr(a) for a in args)
    if sinfo_args is None:
        sinfo_args = []
    if isinstance(sinfo_args, py_tuple):  # type: ignore
        sinfo_args = list(sinfo_args)
    elif not isinstance(sinfo_args, list):
        sinfo_args = [sinfo_args]

    sinfo_args = [
        (
            sinfo()
            if callable(sinfo)
            else sinfo.asobject()
            if isinstance(sinfo, ObjectConvertible)
            else sinfo
        )
        for sinfo in sinfo_args
    ]

    is_default = False
    if "attrs_type_key" in kwargs:
        attrs_type_key = kwargs["attrs_type_key"]
        kwargs.pop("attrs_type_key")
    else:
        attrs_type_key = "ir.DictAttrs"
        is_default = True
    attrs = None
    if kwargs or not is_default:
        attrs = tvm.ir.attrs.make_node(attrs_type_key, **kwargs)

    return Call(op, args, attrs=attrs, sinfo_args=sinfo_args)


def call_py_func(