llvm/lib/SandboxIR/Region.cpp

   1 //===- Region.cpp ---------------------------------------------------------===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #include "llvm/SandboxIR/Region.h"
  10 #include "llvm/SandboxIR/Function.h"
  11
  12 namespace llvm::sandboxir {
  13
  14 Region::Region(Context &Ctx) : Ctx(Ctx) {
  15   LLVMContext &LLVMCtx = Ctx.LLVMCtx;
  16   auto *RegionStrMD = MDString::get(LLVMCtx, RegionStr);
  17   RegionMDN = MDNode::getDistinct(LLVMCtx, {RegionStrMD});
  18
  19   CreateInstCB = Ctx.registerCreateInstrCallback(
  20       [this](Instruction *NewInst) { add(NewInst); });
  21   EraseInstCB = Ctx.registerEraseInstrCallback(
  22       [this](Instruction *ErasedInst) { remove(ErasedInst); });
  23 }
  24
  25 Region::~Region() {
  26   Ctx.unregisterCreateInstrCallback(CreateInstCB);
  27   Ctx.unregisterEraseInstrCallback(EraseInstCB);
  28 }
  29
  30 void Region::add(Instruction *I) {
  31   Insts.insert(I);
  32   // TODO: Consider tagging instructions lazily.
  33   cast<llvm::Instruction>(I->Val)->setMetadata(MDKind, RegionMDN);
  34 }
  35
  36 void Region::remove(Instruction *I) {
  37   Insts.remove(I);
  38   cast<llvm::Instruction>(I->Val)->setMetadata(MDKind, nullptr);
  39 }
  40
  41 #ifndef NDEBUG
  42 bool Region::operator==(const Region &Other) const {
  43   if (Insts.size() != Other.Insts.size())
  44     return false;
  45   if (!std::is_permutation(Insts.begin(), Insts.end(), Other.Insts.begin()))
  46     return false;
  47   return true;
  48 }
  49
  50 void Region::dump(raw_ostream &OS) const {
  51   for (auto *I : Insts)
  52     OS << *I << "\n";
  53 }
  54
  55 void Region::dump() const {
  56   dump(dbgs());
  57   dbgs() << "\n";
  58 }
  59 #endif // NDEBUG
  60
  61 SmallVector<std::unique_ptr<Region>> Region::createRegionsFromMD(Function &F) {
  62   SmallVector<std::unique_ptr<Region>> Regions;
  63   DenseMap<MDNode *, Region *> MDNToRegion;
  64   auto &Ctx = F.getContext();
  65   for (BasicBlock &BB : F) {
  66     for (Instruction &Inst : BB) {
  67       if (auto *MDN = cast<llvm::Instruction>(Inst.Val)->getMetadata(MDKind)) {
  68         Region *R = nullptr;
  69         auto It = MDNToRegion.find(MDN);
  70         if (It == MDNToRegion.end()) {
  71           Regions.push_back(std::make_unique<Region>(Ctx));
  72           R = Regions.back().get();
  73           MDNToRegion[MDN] = R;
  74         } else {
  75           R = It->second;
  76         }
  77         R->add(&Inst);
  78       }
  79     }
  80   }
  81   return Regions;
  82 }
  83
  84 } // namespace llvm::sandboxir