Class Analyzer

python/tvm/arith/analyzer.py:120–426

Integer arithmetic analyzer This is a stateful analyzer class that can be used to perform various symbolic integer analysis. The same analyzer instance can be passed to FFI APIs to share accumulated facts across calls.

@tvm_ffi.register_object("arith.Analyzer")
class Analyzer(Object):
    """Integer arithmetic analyzer

    This is a stateful analyzer class that can be used to perform
    various symbolic integer analysis. The same analyzer instance can
    be passed to FFI APIs to share accumulated facts across calls.
    """

    def __init__(self):
        self.__init_handle_by_constructor__(_ffi_api.Analyzer)

    def const_int_bound(self, expr: tirx.PrimExpr) -> ConstIntBound:
        """Find constant integer bound for expr.

        Parameters
        ----------
        expr : PrimExpr
            The expression.

        Returns
        -------
        bound : ConstIntBound
            The result bound
        """
        return _ffi_api.AnalyzerConstIntBound(self, expr)

    def const_int_bound_is_bound(self, var: tirx.Var) -> bool:
        """Check if a variable is bound to a range.

        Parameters
        ----------
        var : tvm.tirx.Var
            The variable.

        Returns
        -------
        result : bool
            Whether the variable is bound to a range.
        """
        return _ffi_api.AnalyzerConstIntBoundIsBound(self, var)

    def modular_set(self, expr: tirx.PrimExpr) -> ModularSet:
        """Find a modular set that expr belongs to.

        Parameters
        ----------
        expr : PrimExpr
            The expression.

        Returns
        -------
        result : ModularSet
            The result.
        """
        return _ffi_api.AnalyzerModularSet(self, expr)

    def simplify(self, expr: tirx.PrimExpr, steps: int = 2) -> tirx.PrimExpr:
        """Simplify expression via both rewrite and canonicalization.