diff options
Diffstat (limited to 'flang/lib/Optimizer/Analysis')
| -rw-r--r-- | flang/lib/Optimizer/Analysis/TBAAForest.cpp | 15 |
1 files changed, 13 insertions, 2 deletions
diff --git a/flang/lib/Optimizer/Analysis/TBAAForest.cpp b/flang/lib/Optimizer/Analysis/TBAAForest.cpp index cce50e0de1bc..44a0348da3a6 100644 --- a/flang/lib/Optimizer/Analysis/TBAAForest.cpp +++ b/flang/lib/Optimizer/Analysis/TBAAForest.cpp @@ -11,12 +11,23 @@ mlir::LLVM::TBAATagAttr fir::TBAATree::SubtreeState::getTag(llvm::StringRef uniqueName) const { - std::string id = (parentId + "/" + uniqueName).str(); + std::string id = (parentId + '/' + uniqueName).str(); mlir::LLVM::TBAATypeDescriptorAttr type = mlir::LLVM::TBAATypeDescriptorAttr::get( context, id, mlir::LLVM::TBAAMemberAttr::get(parent, 0)); return mlir::LLVM::TBAATagAttr::get(type, type, 0); - // return tag; +} + +fir::TBAATree::SubtreeState & +fir::TBAATree::SubtreeState::getOrCreateNamedSubtree(mlir::StringAttr name) { + auto it = namedSubtrees.find(name); + if (it != namedSubtrees.end()) + return it->second; + + return namedSubtrees + .insert( + {name, SubtreeState(context, parentId + '/' + name.str(), parent)}) + .first->second; } mlir::LLVM::TBAATagAttr fir::TBAATree::SubtreeState::getTag() const { |