Method matches

backends/mlx/pattern_utils.py:156–178

Check if this step fully matches the given node.

(self, node: Node)

    arg_index: int = 0

    def matches(self, node: Node) -> bool:
        """Check if this step fully matches the given node."""
        # Check op or predicate
        if self.op is not None:
            if not match_target(node, self.op):
                return False
        elif self.predicate is not None:
            if not self.predicate(node):
                return False
        else:
            return False

        # Check single user requirement
        if self.require_single_user and not has_single_user(node):
            return False

        # Check nargs and kwargs
        if not self._check_nargs(node):
            return False
        if not self._check_kwargs(node):
            return False

        return True

    def _check_nargs(self, node: Node) -> bool:
        """Check if node has the required number of args."""