как убрать второстепенную ветку в LLVM? - PullRequest
Новая
       91

как убрать второстепенную ветку в LLVM?

1 голос
/ 07 февраля 2020

Я хочу удалить избыточную безусловную ветвь из функции. В следующем примере я хочу удалить br label %26 и объединить их в один базовый c блок. 

; <label>:9:                                      ; preds = %7
  %10 = fadd float %5, %8
  %11 = fmul float %5, %8
  %12 = fadd float %10, %11
  %13 = fdiv float %5, %8
  %14 = fadd float %13, %12
  br label %15

; <label>:15:                                     ; preds = %9
  br label %26

Я пытался сделать это, 

for(auto it1 = F.begin(); it1 != F.end(); it1++){
    BasicBlock& bb = *it1;
    auto BI = dyn_cast<BranchInst>(bb.getTerminator());
    if(BI && BI->isUnconditional() && bb.size() == 1){

        for (BasicBlock *pred : predecessors(&bb)) {
            auto predBI = dyn_cast<BranchInst>(pred->getTerminator());
            if(predBI && predBI->isUnconditional()){
                    predBI->setSuccessor(0, bb.getSingleSuccessor());
                    BI->dropAllReferences();
                    BI->removeFromParent();
            }
        }
    } 

}

Но это дает мне ошибку. Я использую LLVM 6.0.0 

#0 0x000056166a0bcfba (opt+0x11fbfba)
#1 0x000056166a0bb01e (opt+0x11fa01e)
#2 0x000056166a0bb16c (opt+0x11fa16c)
#3 0x00007f2d009cb890 __restore_rt (/lib/x86_64-linux-gnu/libpthread.so.0+0x12890)
#4 0x0000561669b8ce4c (opt+0xccbe4c)
#5 0x00007f2cff43bdb1 mergeBlocks /home/charitha/workspace/rocm/hcc/compiler/lib/Transforms/lcs/MergePass.cpp:103:0
#6 0x00007f2cff43bdb1 (anonymous namespace)::SkeletonPass::runOnFunction(llvm::Function&) /home/charitha/workspace/rocm/hcc/compiler/lib/Transforms/lcs/MergePass.cpp:51:0
#7 0x0000561669bb4f5a (opt+0xcf3f5a)
#8 0x0000561669bb5003 (opt+0xcf4003)
#9 0x0000561669bb4b14 (opt+0xcf3b14)
#10 0x000056166924c765 (opt+0x38b765)
#11 0x00007f2cff65fb97 __libc_start_main (/lib/x86_64-linux-gnu/libc.so.6+0x21b97)
#12 0x000056166929f5fa (opt+0x3de5fa)

Ответы [ 2 ]

1 голос
/ 07 февраля 2020

Вас может заинтересовать llvm::MergeBlockIntoPredecessor() служебная функция, определенная в файле llvm/Transform/Utils/BasicBlockUtils.h.

См. https://llvm.org/doxygen/BasicBlockUtils_8h.html

0 голосов
/ 07 февраля 2020

Я повторно использовал код в llvm/Transform/Utils/BasicBlockUtils.h, и он работал. 

for(auto it1 = F.begin(); it1 != F.end(); it1++){
    BasicBlock* BB = &*it1;
    auto BI = dyn_cast<BranchInst>(BB->getTerminator());
    if(BI && BI->isUnconditional() && BB->size() == 1){
        // code taken from BasicBlockUtils
        auto PredBB = BB->getUniquePredecessor();
        if(PredBB){
            PredBB->getInstList().pop_back();
            BB->replaceAllUsesWith(PredBB);
            PredBB->getInstList().splice(PredBB->end(), BB->getInstList());

            if (!PredBB->hasName())
                PredBB->takeName(BB);

            it1 = BB->eraseFromParent();

        }
    } 
}
...