Class GlobalVar

python/tvm/ir/expr.py:66–108

A global variable in the IR. GlobalVar is used to refer to the global functions stored in the IRModule. Parameters ---------- name_hint: str The name of the variable.

@tvm_ffi.register_object("ir.GlobalVar")
class GlobalVar(RelaxExpr):
    """A global variable in the IR.

    GlobalVar is used to refer to the global functions
    stored in the IRModule.

    Parameters
    ----------
    name_hint: str
        The name of the variable.
    """

    name_hint: str

    def __init__(self, name_hint: str):
        self.__init_handle_by_constructor__(_ffi_api.GlobalVar, name_hint)

    def __call__(self, *args: RelaxExpr) -> BaseExpr:
        """Call the global variable.

        Parameters
        ----------
        args: List[RelaxExpr]
            The arguments to the call.

        Returns
        -------
        call: BaseExpr
            A call taking the variable as a function.
        """
        # pylint: disable=import-outside-toplevel

        # TODO(@relax-team): replace with Relax base class after it's introduced
        if all(isinstance(x, RelaxExpr) for x in args):
            from tvm import relax

            return relax.Call(self, args)

        elif all(isinstance(x, Number | PrimExpr) for x in args):
            return tvm.tirx.call_tir(self, *args)

        arg_types = [type(x) for x in args]
        raise RuntimeError(f"Do not know how to handle GlobalVar.__call__ for types {arg_types}")


@tvm_ffi.register_object("ir.Range")