Function register_patterns

python/tvm/relax/backend/pattern_registry.py:63–85

Register patterns which will be used to partition the DataflowBlock into subgraphs that are supported by external backends. Parameters ---------- patterns: List[Pattern] Patterns to be registered. Patterns that appear later in the list have higher priority when

(patterns: list[Pattern])

def register_patterns(patterns: list[Pattern]):
    """
    Register patterns which will be used to partition the DataflowBlock into
    subgraphs that are supported by external backends.

    Parameters
    ----------
    patterns: List[Pattern]
        Patterns to be registered. Patterns that appear later in the list have
        higher priority when partitioning DataflowBlock.
    """
    _ensure_cleanup_function_registered()

    entries = []
    for item in patterns:
        if isinstance(item, FusionPattern):
            entries.append(item)
        elif isinstance(item, tuple):
            entries.append(FusionPattern(*item))
            _REGISTERED_PATTERN_NAMES.add(item[0])
        else:
            raise TypeError(f"Cannot register type {type(item)} as pattern")
    _ffi_api.RegisterPatterns(entries)


def get_patterns_with_prefix(prefix: str) -> list[FusionPattern]: