shylie/llvm-project
1
0
Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity

Support constructing DominanceInfo with an Operation. This computes the dominance information for any nested regions within the operation.

Browse Source 
PiperOrigin-RevId: 251896520
This commit is contained in:
River Riddle 2019-06-06 11:54:14 -07:00 committed by Mehdi Amini
parent f55f7dc769
commit fa187e0f3b
2 changed files with 23 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mlir/include/mlir/Analysis/Dominance.h
View File

@ -27,6 +27,7 @@ extern template class llvm::DominatorTreeBase<mlir::Block, true>;
namespace mlir {
using DominanceInfoNode = llvm::DomTreeNodeBase<Block>;
class Function;
class Operation;
namespace detail {
template <bool IsPostDom> class DominanceInfoBase {
@ -34,14 +35,16 @@ template <bool IsPostDom> class DominanceInfoBase {
public:
DominanceInfoBase(Function *function) { recalculate(function); }
DominanceInfoBase(Operation *op) { recalculate(op); }
DominanceInfoBase(DominanceInfoBase &&) = default;
DominanceInfoBase &operator=(DominanceInfoBase &&) = default;
DominanceInfoBase(const DominanceInfoBase &) = delete;
DominanceInfoBase &operator=(const DominanceInfoBase &) = delete;
/// Recalculate the dominance info for the provided function.
/// Recalculate the dominance info.
void recalculate(Function *function);
void recalculate(Operation *op);
/// Get the root dominance node of the given region.
DominanceInfoNode *getRootNode(Region *region) {

20
mlir/lib/Analysis/Dominance.cpp
View File

@ -23,6 +23,7 @@
#include "mlir/Analysis/Dominance.h"
#include "mlir/IR/Operation.h"
#include "llvm/Support/GenericDomTreeConstruction.h"
using namespace mlir;
using namespace mlir::detail;
@ -34,7 +35,7 @@ template class llvm::DomTreeNodeBase<Block>;
// DominanceInfoBase
//===----------------------------------------------------------------------===//
/// Recalculate the dominance info for the provided function.
/// Recalculate the dominance info.
template <bool IsPostDom>
void DominanceInfoBase<IsPostDom>::recalculate(Function *function) {
dominanceInfos.clear();
@ -58,6 +59,23 @@ void DominanceInfoBase<IsPostDom>::recalculate(Function *function) {
});
}
template <bool IsPostDom>
void DominanceInfoBase<IsPostDom>::recalculate(Operation *op) {
dominanceInfos.clear();
/// Build the dominance for each of the operation regions.
op->walk([&](Operation *op) {
for (auto &region : op->getRegions()) {
// Don't compute dominance if the region is empty.
if (region.empty())
continue;
auto opDominance = llvm::make_unique<base>();
opDominance->recalculate(region);
dominanceInfos.try_emplace(&region, std::move(opDominance));
}
});
}
/// Return true if the specified block A properly dominates block B.
template <bool IsPostDom>
bool DominanceInfoBase<IsPostDom>::properlyDominates(Block *a, Block *b) {