configure via transaction ops (Xilinx#1682)

fifield · Aug 20, 2024 · f07a0a1 · f07a0a1
1 parent 4629dcd
commit f07a0a1
Show file tree

Hide file tree

Showing 21 changed files with 901 additions and 85 deletions.
diff --git a/include/aie-c/Translation.h b/include/aie-c/Translation.h
@@ -30,6 +30,10 @@ aieTranslateToCDODirect(MlirOperation moduleOp, MlirStringRef workDirPath,
                         bool bigEndian, bool emitUnified, bool cdoDebug,
                         bool aieSim, bool xaieDebug, bool enableCores);
 
+MLIR_CAPI_EXPORTED MlirLogicalResult
+aieTranslateToTxn(MlirOperation moduleOp, MlirStringRef workDirPath,
+                  bool aieSim, bool xaieDebug, bool enableCores);
+
 #ifdef __cplusplus
 }
 #endif

diff --git a/include/aie/Targets/AIETargets.h b/include/aie/Targets/AIETargets.h
@@ -51,6 +51,12 @@ AIETranslateToCDODirect(mlir::ModuleOp m, llvm::StringRef workDirPath,
                         bool bigEndian = false, bool emitUnified = false,
                         bool cdoDebug = false, bool aieSim = false,
                         bool xaieDebug = false, bool enableCores = true);
+mlir::LogicalResult AIETranslateToTxn(mlir::ModuleOp m,
+                                      llvm::StringRef workDirPath,
+                                      bool aieSim = false,
+                                      bool xaieDebug = false,
+                                      bool enableCores = true);
+
 #ifdef AIE_ENABLE_AIRBIN
 mlir::LogicalResult AIETranslateToAirbin(mlir::ModuleOp module,
                                          const std::string &outputFilename,

diff --git a/lib/CAPI/Translation.cpp b/lib/CAPI/Translation.cpp
@@ -79,6 +79,25 @@ MlirLogicalResult aieTranslateToCDODirect(MlirOperation moduleOp,
   return wrap(status);
 }
 
+MlirLogicalResult aieTranslateToTxn(MlirOperation moduleOp,
+                                    MlirStringRef workDirPath, bool aieSim,
+                                    bool xaieDebug, bool enableCores) {
+  ModuleOp mod = llvm::cast<ModuleOp>(unwrap(moduleOp));
+  auto status = AIETranslateToTxn(
+      mod, llvm::StringRef(workDirPath.data, workDirPath.length), aieSim,
+      xaieDebug, enableCores);
+  std::vector<std::string> diagnostics;
+  ScopedDiagnosticHandler handler(mod.getContext(), [&](Diagnostic &d) {
+    llvm::raw_string_ostream(diagnostics.emplace_back())
+        << d.getLocation() << ": " << d;
+  });
+  if (failed(status))
+    for (const auto &diagnostic : diagnostics)
+      std::cerr << diagnostic << "\n";
+
+  return wrap(status);
+}
+
 MlirStringRef aieTranslateToNPU(MlirOperation moduleOp) {
   std::string npu;
   llvm::raw_string_ostream os(npu);