Commit 99ea8ac9 authored by Johannes Doerfert's avatar Johannes Doerfert
Browse files

Reapply "[OpenMP] Group side-effects to improve guarding efficiency" 

This reapplies ca134c39, effectively
reverting commit d2f206e0.

Minor test changes to make the test pass.
parent c09fbbdc

llvm/lib/Transforms/IPO/OpenMPOpt.cpp

+28 −1
Original line number Diff line number Diff line
@@ -3188,6 +3188,34 @@ struct AAKernelInfoFunction : AAKernelInfo { 
            ->setDebugLoc(DL);

    };



    auto &AllocSharedRFI = OMPInfoCache.RFIs[OMPRTL___kmpc_alloc_shared];

    SmallPtrSet<BasicBlock *, 8> Visited;

    for (Instruction *GuardedI : SPMDCompatibilityTracker) {

      BasicBlock *BB = GuardedI->getParent();

      if (!Visited.insert(BB).second)

        continue;



      SmallVector<std::pair<Instruction *, Instruction *>> Reorders;

      Instruction *LastEffect = nullptr;

      BasicBlock::reverse_iterator IP = BB->rbegin(), IPEnd = BB->rend();

      while (++IP != IPEnd) {

        if (!IP->mayHaveSideEffects() && !IP->mayReadFromMemory())

          continue;

        Instruction *I = &*IP;

        if (OpenMPOpt::getCallIfRegularCall(*I, &AllocSharedRFI))

          continue;

        if (!I->user_empty() || !SPMDCompatibilityTracker.contains(I)) {

          LastEffect = nullptr;

          continue;

        }

        if (LastEffect)

          Reorders.push_back({I, LastEffect});

        LastEffect = &*IP;

      }

      for (auto &Reorder : Reorders)

        Reorder.first->moveBefore(Reorder.second);

    }



    SmallVector<std::pair<Instruction *, Instruction *>, 4> GuardedRegions;



    for (Instruction *GuardedI : SPMDCompatibilityTracker) {
@@ -4244,7 +4272,6 @@ void OpenMPOpt::registerAAs(bool IsModulePass) { 
          DepClassTy::NONE, /* ForceUpdate */ false,

          /* UpdateAfterInit */ false);





    registerFoldRuntimeCall(OMPRTL___kmpc_is_generic_main_thread_id);

    registerFoldRuntimeCall(OMPRTL___kmpc_is_spmd_exec_mode);

    registerFoldRuntimeCall(OMPRTL___kmpc_parallel_level);

llvm/test/Transforms/OpenMP/spmdization_guarding.ll

0 → 100644
+420 −0

File added.

Preview size limit exceeded, changes collapsed.