fix shared-library build (#348)

Move a table from being a static variable in the dependee lib into a
function returning a reference in the dependence lib.

Link enough stuff in CMake.

Author: ftynse

lib/polygeist/Ops.cpp

@@ -494,7 +494,16 @@ class BarrierHoist final : public OpRewritePattern<BarrierOp> {
   }
 };
 
-extern std::set<std::string> NonCapturingFunctions;
+const std::set<std::string> &getNonCapturingFunctions() {
+  static std::set<std::string> NonCapturingFunctions = {
+      "free",           "printf",       "fprintf",       "scanf",
+      "fscanf",         "gettimeofday", "clock_gettime", "getenv",
+      "strrchr",        "strlen",       "sprintf",       "sscanf",
+      "mkdir",          "fwrite",       "fread",         "memcpy",
+      "cudaMemcpy",     "memset",       "cudaMemset",    "__isoc99_scanf",
+      "__isoc99_fscanf"};
+  return NonCapturingFunctions;
+}
 
 bool isCaptured(Value v, Operation *potentialUser = nullptr,
                 bool *seenuse = nullptr) {
@@ -560,12 +569,12 @@ bool isCaptured(Value v, Operation *potentialUser = nullptr,
       }
       if (auto cop = dyn_cast<LLVM::CallOp>(u)) {
         if (auto callee = cop.getCallee()) {
-          if (NonCapturingFunctions.count(callee->str()))
+          if (getNonCapturingFunctions().count(callee->str()))
             continue;
         }
       }
       if (auto cop = dyn_cast<func::CallOp>(u)) {
-        if (NonCapturingFunctions.count(cop.getCallee().str()))
+        if (getNonCapturingFunctions().count(cop.getCallee().str()))
           continue;
       }
       return true;

lib/polygeist/Passes/CMakeLists.txt

@@ -32,6 +32,9 @@ add_mlir_dialect_library(MLIRPolygeistTransforms
   MLIRFuncDialect
   MLIRFuncTransforms
   MLIRGPUOps
+  MLIRGPUToGPURuntimeTransforms
+  MLIRGPUTransforms
+  MLIRGPUToNVVMTransforms
   MLIRIR
   MLIRLLVMDialect
   MLIRMathDialect

lib/polygeist/Passes/Mem2Reg.cpp

@@ -1066,13 +1066,8 @@ void removeRedundantBlockArgs(
   }
 }
 
-std::set<std::string> NonCapturingFunctions = {
-    "free",           "printf",       "fprintf",       "scanf",
-    "fscanf",         "gettimeofday", "clock_gettime", "getenv",
-    "strrchr",        "strlen",       "sprintf",       "sscanf",
-    "mkdir",          "fwrite",       "fread",         "memcpy",
-    "cudaMemcpy",     "memset",       "cudaMemset",    "__isoc99_scanf",
-    "__isoc99_fscanf"};
+const std::set<std::string> &getNonCapturingFunctions();
+
 // fopen, fclose
 std::set<std::string> NoWriteFunctions = {"exit", "__errno_location"};
 // This is a straightforward implementation not optimized for speed. Optimize
@@ -1241,7 +1236,7 @@ bool Mem2Reg::forwardStoreToLoad(
         if (callOp.getCallee() != "free") {
           LLVM_DEBUG(llvm::dbgs() << "Aliasing Store: " << callOp << "\n");
           AliasingStoreOperations.insert(callOp);
-          if (!NonCapturingFunctions.count(callOp.getCallee().str()))
+          if (!getNonCapturingFunctions().count(callOp.getCallee().str()))
             captured = true;
         }
         continue;
@@ -1251,7 +1246,7 @@ bool Mem2Reg::forwardStoreToLoad(
           LLVM_DEBUG(llvm::dbgs() << "Aliasing Store: " << callOp << "\n");
           AliasingStoreOperations.insert(callOp);
           if (!callOp.getCallee() ||
-              !NonCapturingFunctions.count(callOp.getCallee()->str()))
+              !getNonCapturingFunctions().count(callOp.getCallee()->str()))
             captured = true;
         }
         continue;
@@ -1825,11 +1820,11 @@ bool isPromotable(mlir::Value AI) {
       } else if (isa<memref::DeallocOp>(U)) {
         continue;
       } else if (auto callOp = dyn_cast<func::CallOp>(U)) {
-        if (NonCapturingFunctions.count(callOp.getCallee().str()))
+        if (getNonCapturingFunctions().count(callOp.getCallee().str()))
           continue;
       } else if (auto callOp = dyn_cast<LLVM::CallOp>(U)) {
         if (auto callee = callOp.getCallee())
-          if (NonCapturingFunctions.count(callee->str()))
+          if (getNonCapturingFunctions().count(callee->str()))
             continue;
       } else if (auto CO = dyn_cast<memref::CastOp>(U)) {
         list.push_back(CO);