[OpenMP] Introduce RAII to protect certain RTL calls from DCE

[jhuber6]

Summary

This patch introduces a new RAII struct that will temporarily make an OpenMP
RTL function have external linkage. This is done before the attributor is
invoked to prevent it from incorrectly removing some function definitions that
we will use later. For example, if we determine all calls to one function are
dead, because it has internal linkage it can safely be removed. Later when we
try to get an instance to that function to modify the source using
getOrCreateRuntimeFunction we will then get an empty declaration for that
function that won't be defined anywhere. This patch prevents this from
occurring.

[jdoerfert]

LG, two nits.

llvm/lib/Transforms/IPO/OpenMPOpt.cpp

1787

no need to store OMPInfoCache

1818–1821

_tgt_... are not device runtime functions, no need to have them here.

[jdoerfert]

llvm/lib/Transforms/IPO/OpenMPOpt.cpp

1818–1821

You need to create objects with a lifetime that matches the scope. You are creating unnamed objects here that die right away.

[jhuber6]

Fixing RAII and removing unnecessary functions.

llvm/lib/Transforms/IPO/OpenMPOpt.cpp

if (RemarkName.startswith("OMP"))
         return RemarkCB(RemarkKind(DEBUG_TYPE, RemarkName, F))
                << " [" << RemarkName << "]";
       });
     else
       ORE.emit(
           [&]() { return RemarkCB(RemarkKind(DEBUG_TYPE, RemarkName, F)); });
   }
 
+  /// RAII struct to temporarily change an RTL function's linkage to external.
+  /// This prevents it from being mistakenly removed by other optimizations.
+  struct ExternalizationRAII {
+    ExternalizationRAII(OMPInformationCache &OMPInfoCache,
+                        RuntimeFunction RFKind)
+        : OMPInfoCache(OMPInfoCache),
+          Declaration(OMPInfoCache.RFIs[RFKind].Declaration) {
+      if (!Declaration)
+        return;
+
+      LinkageType = Declaration->getLinkage();
+      Declaration->setLinkage(GlobalValue::ExternalLinkage);
+    }
+
+    ~ExternalizationRAII() {
+      if (!Declaration)
+        return;
+
+      Declaration->setLinkage(LinkageType);
+    }
+
+    OMPInformationCache &OMPInfoCache;
+    Function *Declaration;
+    GlobalValue::LinkageTypes LinkageType;
+  };
+
   /// The underlying module.
   Module &M;
 
   /// The SCC we are operating on.
   SmallVectorImpl<Function *> &SCC;
 
   /// Callback to update the call graph, the first argument is a removed call,
   /// the second an optional replacement call.
   CallGraphUpdater &CGUpdater;
 
   /// Callback to get an OptimizationRemarkEmitter from a Function *
   OptimizationRemarkGetter OREGetter;
 
   /// OpenMP-specific information cache. Also Used for Attributor runs.
   OMPInformationCache &OMPInfoCache;
 
   /// Attributor instance.
   Attributor &A;
 
   /// Helper function to run Attributor on SCC.
   bool runAttributor(bool IsModulePass) {
     if (SCC.empty())
       return false;
 
+    // Temporarily make these function have external linkage so the Attributor
+    // doesn't remove them when we try to look them up later.
+    ExternalizationRAII(OMPInfoCache, OMPRTL___kmpc_kernel_end_parallel);
+    ExternalizationRAII(OMPInfoCache, OMPRTL___kmpc_kernel_parallel);
+    ExternalizationRAII(OMPInfoCache, OMPRTL___kmpc_barrier_simple_spmd);
+    ExternalizationRAII(OMPInfoCache,
+                        OMPRTL___tgt_target_data_begin_mapper_wait);
+    ExternalizationRAII(OMPInfoCache,
+                        OMPRTL___tgt_target_data_begin_mapper_issue);
+
     registerAAs(IsModulePass);
 
     ChangeStatus Changed = A.run();
 
     LLVM_DEBUG(dbgs() << "[Attributor] Done with " << SCC.size()
                       << " functions, result: " << Changed << ".\n");
 
     return Changed == ChangeStatus::CHANGED;