done!

llvm · Jun 14, 2024 · d4e6e05 · d4e6e05
1 parent 4745dae
commit d4e6e05
Show file tree

Hide file tree

Showing 5 changed files with 13 additions and 2 deletions.
diff --git a/lib/Conversion/PolygeistToLLVM/CMakeLists.txt b/lib/Conversion/PolygeistToLLVM/CMakeLists.txt
@@ -21,3 +21,10 @@ add_mlir_conversion_library(MLIRPolygeistToLLVM
   MLIRSCFToControlFlow
   MLIRVectorToLLVM
 )
+
+target_compile_definitions(obj.MLIRPolygeistToLLVM
+  PRIVATE
+  POLYGEIST_PGO_DEFAULT_DATA_DIR="${POLYGEIST_PGO_DEFAULT_DATA_DIR}"
+  POLYGEIST_PGO_ALTERNATIVE_ENV_VAR="${POLYGEIST_PGO_ALTERNATIVE_ENV_VAR}"
+  POLYGEIST_PGO_DATA_DIR_ENV_VAR="${POLYGEIST_PGO_DATA_DIR_ENV_VAR}"
+)
diff --git a/lib/Conversion/PolygeistToLLVM/PolygeistToLLVM.cpp b/lib/Conversion/PolygeistToLLVM/PolygeistToLLVM.cpp
@@ -10,6 +10,8 @@
 // a generic parallel for representation
 //===----------------------------------------------------------------------===//
 
+#include "PassDetails.h"
+
 #include "mlir/../../lib/Conversion/MemRefToLLVM/MemRefToLLVM.cpp"
 #include "mlir/Analysis/DataLayoutAnalysis.h"
 #include "mlir/Conversion/ArithToLLVM/ArithToLLVM.h"

diff --git a/tools/cgeist/CMakeLists.txt b/tools/cgeist/CMakeLists.txt
@@ -59,6 +59,7 @@ target_compile_definitions(cgeist PUBLIC -DLLVM_OBJ_ROOT="${LLVM_BINARY_DIR}")
 target_link_libraries(cgeist PRIVATE
   MLIRSCFTransforms
   MLIRPolygeistDialect
+  MLIRPolygeistToLLVM
 
   MLIRSupport
   MLIRIR

diff --git a/tools/polygeist-opt/CMakeLists.txt b/tools/polygeist-opt/CMakeLists.txt
@@ -4,7 +4,8 @@ set(LIBS
         ${dialect_libs}
         ${conversion_libs}
         MLIROptLib
-		MLIRPolygeist
+		MLIRPolygeistDialect
+		MLIRPolygeistToLLVM
 		MLIRPolygeistTransforms
 		MLIRFuncAllExtensions
         )

diff --git a/tools/polygeist-opt/polygeist-opt.cpp b/tools/polygeist-opt/polygeist-opt.cpp
@@ -62,7 +62,7 @@ int main(int argc, char **argv) {
   registry.insert<mlir::polygeist::PolygeistDialect>();
   registry.insert<DLTIDialect>();
 
-  mlir::registerpolygeistPasses();
+  mlir::registerPolygeistPasses();
   mlir::func::registerInlinerExtension(registry);
 
   // Register the standard passes we want.