Class IntImm

python/tvm/tirx/expr.py:611–652

Int constant. Parameters ---------- dtype : str The data type value : int The constant value. span : Optional[Span] The location of this expression in the source code.

@tvm_ffi.register_object("ir.IntImm")
class IntImm(ConstExpr):
    """Int constant.

    Parameters
    ----------
    dtype : str
        The data type

    value : int
        The constant value.

    span : Optional[Span]
        The location of this expression in the source code.
    """

    value: int

    def __init__(self, dtype: str, value: int, span: Span | None = None) -> None:
        self.__init_handle_by_constructor__(
            tvm.ir._ffi_api.IntImm,
            dtype,
            value,
            span,  # type: ignore
        )

    def __hash__(self) -> int:
        return self.value

    def __int__(self) -> int:
        return self.value

    def __nonzero__(self) -> bool:
        return self.value != 0

    def __eq__(self, other: PrimExpr) -> PrimExpr:
        return _ffi_api._OpEQ(self, other, None)  # type: ignore

    def __ne__(self, other: PrimExpr) -> PrimExpr:
        return _ffi_api._OpNE(self, other, None)  # type: ignore

    def __bool__(self) -> bool:
        return self.__nonzero__()


@tvm_ffi.register_object("tirx.StringImm")  # type: ignore