[mlir][python] ExecutionEngine can dump to object file

[bondhugula]

mlir/lib/Bindings/Python/ExecutionEngine.cpp

104

Nit: Is this properly phrased? It's not the engine itself but the the IR being JITted that is being dumped?

mlir/include/mlir-c/ExecutionEngine.h

MLIR_CAPI_EXPORTED void *mlirExecutionEngineLookup(MlirExecutionEngine jit,
                                                    MlirStringRef name);
 
 /// Register a symbol with the jit: this symbol will be accessible to the jitted
 /// code.
 MLIR_CAPI_EXPORTED void
 mlirExecutionEngineRegisterSymbol(MlirExecutionEngine jit, MlirStringRef name,
                                   void *sym);
 
+/// Dump as an object in `fileName`.
+MLIR_CAPI_EXPORTED void
+mlirExecutionEngineDumpToObjectFile(MlirExecutionEngine jit,
+                                    MlirStringRef fileName);
+
 #ifdef __cplusplus
 }
 #endif
 
 #endif // EXECUTIONENGINE_H

mlir/lib/Bindings/Python/ExecutionEngine.cpp

py::class_<PyExecutionEngine>(m, "ExecutionEngine")
           "raw_register_runtime",
           [](PyExecutionEngine &executionEngine, const std::string &name,
              uintptr_t sym) {
             mlirExecutionEngineRegisterSymbol(
                 executionEngine.get(),
                 mlirStringRefCreate(name.c_str(), name.size()),
                 reinterpret_cast<void *>(sym));
           },
-          "Lookup function `func` in the ExecutionEngine.");
+          "Lookup function `func` in the ExecutionEngine.")
+      .def(
+          "dump_to_object_file",
+          [](PyExecutionEngine &executionEngine, const std::string &fileName) {
+            mlirExecutionEngineDumpToObjectFile(
+                executionEngine.get(),
+                mlirStringRefCreate(fileName.c_str(), fileName.size()));
+          },
+          "Dump ExecutionEngine to an object file.");
 }

mlir/lib/CAPI/ExecutionEngine/ExecutionEngine.cpp

extern "C" void mlirExecutionEngineRegisterSymbol(MlirExecutionEngine jit,
                                                   void *sym) {
   unwrap(jit)->registerSymbols([&](llvm::orc::MangleAndInterner interner) {
     llvm::orc::SymbolMap symbolMap;
     symbolMap[interner(unwrap(name))] =
         llvm::JITEvaluatedSymbol::fromPointer(sym);
     return symbolMap;
   });
 }
+
+extern "C" void mlirExecutionEngineDumpToObjectFile(MlirExecutionEngine jit,
+                                                    MlirStringRef name) {
+  unwrap(jit)->dumpToObjectFile(unwrap(name));
+}