Walk.h

Go to the documentation of this file.

//===-- Walk.h --------------------------------------------------*- C++ -*-===//
//
// Part of the LLZK Project, under the Apache License v2.0.
// See LICENSE.txt for license information.
// Copyright 2026 Project LLZK
// SPDX-License-Identifier: Apache-2.0
//
//===----------------------------------------------------------------------===//
 
#pragma once
 
#include <mlir/IR/Visitors.h>
 
#include <llvm/ADT/STLFunctionalExtras.h>
#include <llvm/ADT/SmallVector.h>

template <typename MatchType, typename R>
inline static bool walkContainsMatch(R &root, llvm::function_ref<bool(MatchType)> pred) {
  return root
      .walk([&pred](MatchType t) {
    return pred(t) ? mlir::WalkResult::interrupt() : mlir::WalkResult::advance();
  }).wasInterrupted();
}

template <typename MatchType, typename R> inline static bool walkContains(R &root) {
  return root.walk([](MatchType t) { return mlir::WalkResult::interrupt(); }).wasInterrupted();
}

template <typename MatchType, typename R>
inline static llvm::SmallVector<MatchType> walkCollect(R &root) {
  llvm::SmallVector<MatchType> collected;
  root.walk([&collected](MatchType op) { collected.push_back(op); });
  return collected;
}