1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
//===- Visitors.cpp - MLIR Visitor Utilities ------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#include "mlir/IR/Visitors.h"
#include "mlir/IR/Operation.h"
using namespace mlir;
WalkStage::WalkStage(Operation *op)
: numRegions(op->getNumRegions()), nextRegion(0) {}
MutableArrayRef<Region> ForwardIterator::makeIterable(Operation &range) {
return range.getRegions();
}
void detail::walk(Operation *op,
function_ref<void(Operation *, const WalkStage &)> callback) {
WalkStage stage(op);
for (Region ®ion : op->getRegions()) {
// Invoke callback on the parent op before visiting each child region.
callback(op, stage);
stage.advance();
for (Block &block : region) {
for (Operation &nestedOp : block)
walk(&nestedOp, callback);
}
}
// Invoke callback after all regions have been visited.
callback(op, stage);
}
WalkResult detail::walk(
Operation *op,
function_ref<WalkResult(Operation *, const WalkStage &)> callback) {
WalkStage stage(op);
for (Region ®ion : op->getRegions()) {
// Invoke callback on the parent op before visiting each child region.
WalkResult result = callback(op, stage);
if (result.wasSkipped())
return WalkResult::advance();
if (result.wasInterrupted())
return WalkResult::interrupt();
stage.advance();
for (Block &block : region) {
// Early increment here in the case where the operation is erased.
for (Operation &nestedOp : llvm::make_early_inc_range(block))
if (walk(&nestedOp, callback).wasInterrupted())
return WalkResult::interrupt();
}
}
return callback(op, stage);
}
|
