github.com/johnnyeven/libtools@v0.0.0-20191126065708-61829c1adf46/third_party/mlir/lib/IR/Visitors.cpp (about)

     1  //===- Visitors.cpp - MLIR Visitor Utilties -------------------------------===//
     2  //
     3  // Copyright 2019 The MLIR Authors.
     4  //
     5  // Licensed under the Apache License, Version 2.0 (the "License");
     6  // you may not use this file except in compliance with the License.
     7  // You may obtain a copy of the License at
     8  //
     9  //   http://www.apache.org/licenses/LICENSE-2.0
    10  //
    11  // Unless required by applicable law or agreed to in writing, software
    12  // distributed under the License is distributed on an "AS IS" BASIS,
    13  // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    14  // See the License for the specific language governing permissions and
    15  // limitations under the License.
    16  // =============================================================================
    17  
    18  #include "mlir/IR/Visitors.h"
    19  #include "mlir/IR/Operation.h"
    20  
    21  using namespace mlir;
    22  
    23  /// Walk all of the operations nested under and including the given operations.
    24  void detail::walkOperations(Operation *op,
    25                              function_ref<void(Operation *op)> callback) {
    26    // TODO(b/140235992) This walk should be iterative over the operations.
    27    for (auto &region : op->getRegions())
    28      for (auto &block : region)
    29        // Early increment here in the case where the operation is erased.
    30        for (auto &nestedOp : llvm::make_early_inc_range(block))
    31          walkOperations(&nestedOp, callback);
    32  
    33    callback(op);
    34  }
    35  
    36  /// Walk all of the operations nested under and including the given operations.
    37  /// This methods walks operations until an interrupt signal is received.
    38  WalkResult
    39  detail::walkOperations(Operation *op,
    40                         function_ref<WalkResult(Operation *op)> callback) {
    41    // TODO(b/140235992) This walk should be iterative over the operations.
    42    for (auto &region : op->getRegions()) {
    43      for (auto &block : region) {
    44        // Early increment here in the case where the operation is erased.
    45        for (auto &nestedOp : llvm::make_early_inc_range(block))
    46          if (walkOperations(&nestedOp, callback).wasInterrupted())
    47            return WalkResult::interrupt();
    48      }
    49    }
    50    return callback(op);
    51  }