fix mac windows python shared library loading

Author: drblallo

lib/conversions/src/RLCToPython.cpp

@@ -103,7 +103,7 @@ namespace mlir::rlc
 		std::map<const void*, mlir::rlc::FunctionOp> dropFunction;
 		std::map<const void*, mlir::rlc::FunctionOp> assignFunction;
 	};
-	static void printPrelude(StreamWriter& writer)
+	static void printPrelude(StreamWriter& writer, bool isMac, bool isWindows)
 	{
 		writer.writenl("import ctypes");
 		writer.writenl("import os");
@@ -112,9 +112,16 @@ namespace mlir::rlc
 		writer.writenl("import builtins");
 		writer.writenl("from collections import defaultdict");
 
+		std::string libName = "lib.so";
+		if (isMac)
+			libName = "lib.dylib";
+		if (isWindows)
+			libName = "lib.dll";
 		writer.writenl(
 				"lib = ctypes.CDLL(os.path.join(Path(__file__).resolve().parent, "
-				"\"lib.so\"))");
+				"\"",
+				libName,
+				"\"))");
 		writer.writenl("actions = defaultdict(list)");
 		writer.writenl("wrappers = defaultdict(list)");
 		writer.writenl("signatures = {}");
@@ -854,12 +861,11 @@ namespace mlir::rlc
 		OS.endLine();
 	}
 
-#define GEN_PASS_DEF_NEOPRINTPYTHONPASS
+#define GEN_PASS_DEF_PRINTPYTHONPASS
 #include "rlc/dialect/Passes.inc"
-	struct NeoPrintPythonPass: impl::NeoPrintPythonPassBase<NeoPrintPythonPass>
+	struct PrintPythonPass: impl::PrintPythonPassBase<PrintPythonPass>
 	{
-		using impl::NeoPrintPythonPassBase<
-				NeoPrintPythonPass>::NeoPrintPythonPassBase;
+		using impl::PrintPythonPassBase<PrintPythonPass>::PrintPythonPassBase;
 
 		void runOnOperation() override
 		{
@@ -878,7 +884,7 @@ namespace mlir::rlc
 			matcher.add<AliasToPythonAlias>();
 
 			// emit includes
-			printPrelude(matcher.getWriter());
+			printPrelude(matcher.getWriter(), isMac, isWindows);
 
 			// emit declarations of types
 			for (auto t : ::rlc::postOrderTypes(getOperation()))

lib/dialect/src/Passes.td

@@ -181,11 +181,15 @@ def SerializeRLPass : Pass<"rlc-serialize-ir-pass", "mlir::ModuleOp"> {
   let dependentDialects = ["rlc::RLCDialect"];
 }
 
-def NeoPrintPythonPass : Pass<"rlc-new-python-pass", "mlir::ModuleOp"> {
+def PrintPythonPass : Pass<"rlc-python-pass", "mlir::ModuleOp"> {
   let summary = "prints the python header on the provided stream";
   let options = [
     Option<"OS", "output stream", "llvm::raw_ostream*", /*default=*/"nullptr",
-           "stream on which to print the module">
+           "stream on which to print the module">,
+    Option<"isMac", "true if target platform is mac", "bool", /*default=*/"false",
+           "true if target platform is mac">,
+    Option<"isWindows", "true if target platform is windows", "bool", /*default=*/"false",
+           "true if target platform is windows">,
   ];
   let dependentDialects = ["rlc::RLCDialect"];
 }

lib/driver/src/Driver.cpp

@@ -107,7 +107,8 @@ namespace mlir::rlc
 		if (request == Request::dumpPythonWrapper)
 		{
 			manager.addPass(mlir::rlc::createSortTypeDeclarationsPass());
-			manager.addPass(mlir::rlc::createNeoPrintPythonPass({ OS }));
+			manager.addPass(mlir::rlc::createPrintPythonPass(
+					{ OS, targetInfo->isMacOS(), targetInfo->isWindows() }));
 			return;
 		}