Runtime TXN generation

include/aie/Conversion/AIEXToEmitC/AIEXToEmitC.h

@@ -0,0 +1,25 @@
+//===- AIEXToEmitC.h - AIEX to EmitC conversion -----------------*- C++ -*-===//
+//
+// This file is licensed under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+// (c) Copyright 2026 Advanced Micro Devices, Inc.
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef AIE_CONVERSION_AIEXTOEMITC_AIEXTOEMITC_H
+#define AIE_CONVERSION_AIEXTOEMITC_AIEXTOEMITC_H
+
+#include "mlir/IR/BuiltinOps.h"
+#include "mlir/Pass/Pass.h"
+#include <memory>
+
+namespace xilinx {
+
+std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>>
+createConvertAIEXToEmitCPass();
+
+} // namespace xilinx
+
+#endif // AIE_CONVERSION_AIEXTOEMITC_AIEXTOEMITC_H

include/aie/Conversion/Passes.h

@@ -13,6 +13,7 @@
 
 #include "aie/Conversion/AIEToConfiguration/AIEToConfiguration.h"
 #include "aie/Conversion/AIEVecToLLVM/AIEVecToLLVM.h"
+#include "aie/Conversion/AIEXToEmitC/AIEXToEmitC.h"
 #include "aie/Conversion/PassesEnums.h.inc"
 
 namespace xilinx {

include/aie/Conversion/Passes.td

@@ -104,4 +104,28 @@ def ConvertAIEToControlPackets : Pass<"convert-aie-to-control-packets",
   ];
 }
 
+//===----------------------------------------------------------------------===//
+// AIEXToEmitC
+//===----------------------------------------------------------------------===//
+
+def ConvertAIEXToEmitC : Pass<"convert-aiex-to-emitc", "mlir::ModuleOp"> {
+  let summary = "Convert AIEX dynamic runtime sequence ops to EmitC dialect";
+  let description = [{
+    This pass converts AIEX runtime sequence operations (write32, maskwrite32,
+    sync — including their dynamic operand forms) along with static NPU ops,
+    SCF control flow, and arith operations into EmitC dialect ops. The
+    resulting EmitC IR can be translated to C++ code via translateToCpp()
+    that calls functions from the standalone TxnEncoding.h library to
+    generate TXN binaries at runtime.
+  }];
+  let constructor = "xilinx::createConvertAIEXToEmitCPass()";
+  let dependentDialects = ["mlir::emitc::EmitCDialect",
+                           "mlir::arith::ArithDialect",
+                           "mlir::func::FuncDialect",
+                           "mlir::memref::MemRefDialect",
+                           "mlir::scf::SCFDialect",
+                           "xilinx::AIE::AIEDialect",
+                           "xilinx::AIEX::AIEXDialect"];
+}
+
 #endif // AIE_CONVERSION_PASSES

include/aie/Dialect/AIE/IR/AIEOps.td

@@ -887,7 +887,7 @@ def AIE_DMABDPACKETOp: AIE_Op<"dma_bd_packet", []> {
   }];
 }
 
-def AIE_DMABDOp: AIE_Op<"dma_bd", []> {
+def AIE_DMABDOp: AIE_Op<"dma_bd", [AttrSizedOperandSegments]> {
   let summary = "Declare a dma buffer descriptor op";
   let description = [{
     This operation describes a buffer descriptor for DMA operations. In particular, it specifies
@@ -997,10 +997,36 @@ def AIE_DMABDOp: AIE_Op<"dma_bd", []> {
     counts can be supplied to the `dma_bd` through an optional argument, an array of "tuple-like" attributes
     `bd_pad_layout<const_pad_before, const_pad_after>`, followed by an optional argument `const_val` (default
     is 0). All counts are expressed in multiples of the element width.
+
+    #### Dynamic operands (SSA `i64`)
+
+    `dma_bd` accepts optional SSA `i64` overrides for the offset, length, and the
+    sizes/strides of the data-layout transformation:
+
+      * `dyn_offset` overrides the static `offset` attribute (in element-width units).
+      * `dyn_len` overrides the static `len` attribute (in element-width units).
+      * `dyn_sizes` and `dyn_strides` together override the static `dimensions`
+        attribute. They are *all-or-nothing*: either both are empty (and
+        `dimensions` is used), or both have the same non-empty length and the
+        static `dimensions` attribute must be absent. To reuse a static value in
+        a dimension, the user passes an `arith.constant` SSA value, which the
+        downstream lowering folds back to a constant word field.
+
+    Dynamic operands are restricted to shim NOC tiles because the dynamic
+    lowering emits per-word `npu.write32` ops that target the runtime sequence
+    on the shim. The user is responsible for ensuring the inner-most stride is 1
+    (this cannot be checked statically when expressed via SSA), and for
+    respecting the hardware d0 wrap-size 10-bit limit.
+
+    `pad_dimensions` is incompatible with dynamic operands.
   }];
 
   let arguments = (
     ins AnyRankedOrUnrankedMemRef:$buffer,
+        Optional<I64>:$dyn_offset,
+        Optional<I64>:$dyn_len,
+        Variadic<I64>:$dyn_sizes,
+        Variadic<I64>:$dyn_strides,
         // in multiples of element width (not bytes)
         DefaultValuedAttr<AIEI32Attr, "0">:$offset,
         // in multiples of element width (not bytes)
@@ -1026,6 +1052,8 @@ def AIE_DMABDOp: AIE_Op<"dma_bd", []> {
       return dataLayout.getTypeSize(bufferType.getElementType());
     }
     uint64_t getLenInBytes() {
+      assert(!getDynLen() && "getLenInBytes called on dma_bd with dynamic len; "
+                             "the caller must take the dynamic-lowering path");
       if (std::optional<int32_t> len = getLen(); len.has_value())
         return static_cast<uint64_t>(static_cast<uint32_t>(len.value())) *
                static_cast<uint64_t>(getBufferElementTypeWidthInBytes());
@@ -1046,50 +1074,66 @@ def AIE_DMABDOp: AIE_Op<"dma_bd", []> {
   let builders = [
     OpBuilder<(ins "mlir::Value":$buffer, "int":$offset, "int":$len), [{
       $_state.addOperands(buffer);
+      $_state.addAttribute(getOperandSegmentSizeAttr(),
+          $_builder.getDenseI32ArrayAttr({1, 0, 0, 0, 0}));
       $_state.addAttribute("offset", $_builder.getI32IntegerAttr(offset));
       $_state.addAttribute("len", $_builder.getI32IntegerAttr(len));
     }]>,
     OpBuilder<(ins "mlir::Value":$buffer, "int":$offset, "int":$len, "BDDimLayoutArrayAttr":$dims), [{
       $_state.addOperands(buffer);
+      $_state.addAttribute(getOperandSegmentSizeAttr(),
+          $_builder.getDenseI32ArrayAttr({1, 0, 0, 0, 0}));
       $_state.addAttribute("offset", $_builder.getI32IntegerAttr(offset));
       $_state.addAttribute("len", $_builder.getI32IntegerAttr(len));
       $_state.addAttribute("dimensions", dims);
     }]>,
     OpBuilder<(ins "mlir::Value":$buffer, "int":$offset, "int":$len, "BDPadLayoutArrayAttr":$paddims), [{
       $_state.addOperands(buffer);
+      $_state.addAttribute(getOperandSegmentSizeAttr(),
+          $_builder.getDenseI32ArrayAttr({1, 0, 0, 0, 0}));
       $_state.addAttribute("offset", $_builder.getI32IntegerAttr(offset));
       $_state.addAttribute("len", $_builder.getI32IntegerAttr(len));
       $_state.addAttribute("pad_dimensions", paddims);
     }]>,
     OpBuilder<(ins "mlir::Value":$buffer, "int":$offset, "int":$len, "BDDimLayoutArrayAttr":$dims, "BDPadLayoutArrayAttr":$paddims), [{
       $_state.addOperands(buffer);
+      $_state.addAttribute(getOperandSegmentSizeAttr(),
+          $_builder.getDenseI32ArrayAttr({1, 0, 0, 0, 0}));
       $_state.addAttribute("offset", $_builder.getI32IntegerAttr(offset));
       $_state.addAttribute("len", $_builder.getI32IntegerAttr(len));
       $_state.addAttribute("dimensions", dims);
       $_state.addAttribute("pad_dimensions", paddims);
     }]>,
     OpBuilder<(ins "mlir::Value":$buffer, "int":$offset, "int":$len, "PacketInfoAttr":$pkt), [{
       $_state.addOperands(buffer);
+      $_state.addAttribute(getOperandSegmentSizeAttr(),
+          $_builder.getDenseI32ArrayAttr({1, 0, 0, 0, 0}));
       $_state.addAttribute("offset", $_builder.getI32IntegerAttr(offset));
       $_state.addAttribute("len", $_builder.getI32IntegerAttr(len));
       $_state.addAttribute("packet", pkt);
     }]>,
     OpBuilder<(ins "mlir::Value":$buffer, "int":$offset, "int":$len, "BDDimLayoutArrayAttr":$dims, "PacketInfoAttr":$pkt), [{
       $_state.addOperands(buffer);
+      $_state.addAttribute(getOperandSegmentSizeAttr(),
+          $_builder.getDenseI32ArrayAttr({1, 0, 0, 0, 0}));
       $_state.addAttribute("offset", $_builder.getI32IntegerAttr(offset));
       $_state.addAttribute("len", $_builder.getI32IntegerAttr(len));
       $_state.addAttribute("dimensions", dims);
       $_state.addAttribute("packet", pkt);
     }]>,
     OpBuilder<(ins "mlir::Value":$buffer, "int":$offset, "int":$len, "BDPadLayoutArrayAttr":$paddims, "PacketInfoAttr":$pkt), [{
       $_state.addOperands(buffer);
+      $_state.addAttribute(getOperandSegmentSizeAttr(),
+          $_builder.getDenseI32ArrayAttr({1, 0, 0, 0, 0}));
       $_state.addAttribute("offset", $_builder.getI32IntegerAttr(offset));
       $_state.addAttribute("len", $_builder.getI32IntegerAttr(len));
       $_state.addAttribute("pad_dimensions", paddims);
       $_state.addAttribute("packet", pkt);
     }]>,
     OpBuilder<(ins "mlir::Value":$buffer, "int":$offset, "int":$len, "BDDimLayoutArrayAttr":$dims, "BDPadLayoutArrayAttr":$paddims, "PacketInfoAttr":$pkt), [{
       $_state.addOperands(buffer);
+      $_state.addAttribute(getOperandSegmentSizeAttr(),
+          $_builder.getDenseI32ArrayAttr({1, 0, 0, 0, 0}));
       $_state.addAttribute("offset", $_builder.getI32IntegerAttr(offset));
       $_state.addAttribute("len", $_builder.getI32IntegerAttr(len));
       $_state.addAttribute("dimensions", dims);
@@ -2247,8 +2291,8 @@ def AIE_BDChainOp: AIE_Op<"bd_chain", [Symbol, SkipAccessibilityCheckTrait]> {
 
 def AIE_RuntimeSequenceOp : AIE_Op<"runtime_sequence", [
   Symbol,
-  NoTerminator, 
-  HasParent<"DeviceOp">, 
+  NoTerminator,
+  HasParent<"DeviceOp">,
 ]> {
   let summary = "Program the configuration co-processor of the AI Engine array";
   let description = [{