Class PatternContext

python/tvm/relax/dpl/context.py:27–79

A context object for doing graph (topogical) pattern matching.

class PatternContext(tvm.runtime.Object):
    """A context object for doing graph (topogical) pattern matching."""

    def __init__(self, incremental=False):
        """
        Initialize the PatternContext

        Parameters
        ----------
        incremental : bool, optional
            perform incremental matching based on the recent context, by default False
        """
        self.__init_handle_by_constructor__(ffi.PatternContext, incremental)  # type: ignore

    def __enter__(self):
        """Enter the context"""
        ffi.enter_context(self)  # type: ignore
        return self

    def __exit__(self, exc_type, exc_value, traceback):
        """Exit the context"""
        ffi.exit_context(self)  # type: ignore

    @staticmethod
    def current() -> "PatternContext":
        """
        Get the current context

        Returns
        -------
        PatternContext
            The current context
        """
        return ffi.current_context()  # type: ignore

    def match_dfb(
        self,
        dfb: DataflowBlock,
    ) -> dict[DFPattern, Var]:
        """
        Match a DataflowBlock via a graph of DFPattern and corresponding constraints

        Parameters
        ----------
        dfb : DataflowBlock
            The DataflowBlock to match

        Returns
        -------
        Dict[DFPattern, Var]
            The mapping from DFPattern to matched expression
        """
        return ffi.match_dfb(self, dfb)  # type: ignore