CascadeFlowOp and ConfigureCascadeOp

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

@@ -144,4 +144,15 @@ def BDDimLayoutArrayArrayAttr : ArrayOfAttr<
     /*eltName*/BDDimLayoutArrayAttr.cppClassName
 >;
 
+def CascadeDir: I32EnumAttr<"CascadeDir", "Directions for cascade",
+  [
+    I32EnumAttrCase<"South", 0>,
+    I32EnumAttrCase<"West", 1>,
+    I32EnumAttrCase<"North", 2>,
+    I32EnumAttrCase<"East", 3>
+  ]> {
+
+  let cppNamespace = "xilinx::AIE";
+}
+
 #endif // AIE_ATTRS

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

@@ -1343,6 +1343,30 @@ def AIE_PutStreamOp: AIE_Op<"put_stream", [HasParent<"CoreOp">]> {
   }];
 }
 
+def AIE_ConfigureCascadeOp: AIE_Op<"configure_cascade", [HasParent<"DeviceOp">]> {
+  let summary = "An op to configure the input and output directions of the cascade for a single AIE tile";
+  let description = [{
+    An operation to configure the cascade on a single tile in both the input and the output 
+    directions.
+
+    Example:
+    ```
+      %tile00 = aie.tile(1, 3)
+      aie.configure_cascade(%tile00, West, East)
+    ```
+    Configures the input cascade port of %tile00 to the West direction, and the output port to the East direction.
+  }];
+
+  let arguments = (
+    ins Index:$tile,
+        CascadeDir:$inputDir,
+        CascadeDir:$outputDir
+  );
+  let results = (outs);
+  let hasVerifier = 1;
+  let assemblyFormat = [{ `(` $tile `,` $inputDir `,` $outputDir `)` attr-dict }];
+}
+
 def AIE_GetCascadeOp: AIE_Op<"get_cascade", [HasParent<"CoreOp">]>, Results<(outs AnyType:$cascadeValue)> {
   let summary = "An op to read from a cascading stream from a neighboring core";
   let description = [{

lib/Dialect/AIE/IR/AIEDialect.cpp

@@ -864,6 +864,30 @@ ObjectFifoCreateOp ObjectFifoRegisterProcessOp::getObjectFifo() {
   return {};
 }
 
+//===----------------------------------------------------------------------===//
+// ConfigureCascadeOp
+//===----------------------------------------------------------------------===//
+
+LogicalResult ConfigureCascadeOp::verify() {
+  const auto &targetModel = getTargetModel(*this);
+  CascadeDir inputDir = getInputDir();
+  CascadeDir outputDir = getOutputDir();
+  if (targetModel.getTargetArch() == AIEArch::AIE2) {
+    if (inputDir == CascadeDir::South || inputDir == CascadeDir::East) {
+      return emitOpError("input direction of cascade must be North or West on ")
+             << stringifyAIEArch(targetModel.getTargetArch());
+    }
+    if (outputDir == CascadeDir::North || outputDir == CascadeDir::West) {
+      return emitOpError("output direction of cascade must be South or East on ")
+           << stringifyAIEArch(targetModel.getTargetArch());
+    }
+  } else {
+    return emitOpError("cascade not supported in ")
+           << stringifyAIEArch(targetModel.getTargetArch());
+  }
+  return success();
+}
+
 //===----------------------------------------------------------------------===//
 // PutCascadeOp
 //===----------------------------------------------------------------------===//

lib/Targets/AIETargetXAIEV2.cpp

@@ -765,6 +765,22 @@ mlir::LogicalResult AIETranslateToXAIEV2(ModuleOp module, raw_ostream &output) {
   output << "return XAIE_OK;\n";
   output << "} // mlir_aie_configure_switchboxes\n\n";
 
+  //---------------------------------------------------------------------------
+  // mlir_aie_configure_cascade
+  //---------------------------------------------------------------------------
+  output << "int mlir_aie_configure_cascade(" << ctx_p << ") {\n";
+  for (auto configOp : targetOp.getOps<ConfigureCascadeOp>()) {
+    TileOp tile = dyn_cast<TileOp>(configOp.getTile().getDefiningOp());
+    int col = tile.colIndex();
+    int row = tile.rowIndex();
+    output << "XAie_CoreConfigAccumulatorControl(" << deviceInstRef << ", "
+           << "XAie_TileLoc(" << col << ", " << row << "), "
+           << stringifyCascadeDir(configOp.getInputDir()).upper() << ", "
+           << stringifyCascadeDir(configOp.getOutputDir()).upper() << ");\n";
+  }
+  output << "return XAIE_OK;\n";
+  output << "} // mlir_aie_configure_cascade\n\n";
+
   //---------------------------------------------------------------------------
   // Output Buffer Accessors
   //---------------------------------------------------------------------------

test/dialect/AIE/bad_cascade.mlir

@@ -0,0 +1,35 @@
+//===- bad_cascade.mlir ----------------------------------------*- MLIR -*-===//
+//
+// 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 2024 Advanced Micro Devices, Inc.
+//
+//===----------------------------------------------------------------------===//
+
+// RUN: not aie-opt -split-input-file %s 2>&1 | FileCheck %s
+// CHECK: error{{.*}}'aie.configure_cascade' op input direction of cascade must be North or West on AIE2
+
+aie.device(xcve2802) {
+  %t13 = aie.tile(1, 3)
+  aie.configure_cascade(%t13, East, South)
+}
+
+// -----
+
+// CHECK: error{{.*}}'aie.configure_cascade' op output direction of cascade must be South or East on AIE2
+
+aie.device(xcve2802) {
+  %t13 = aie.tile(1, 3)
+  aie.configure_cascade(%t13, North, West)
+}
+
+// -----
+
+// CHECK: error{{.*}}'aie.configure_cascade' op cascade not supported in AIE1
+
+aie.device(xcvc1902) {
+  %t13 = aie.tile(1, 3)
+  aie.configure_cascade(%t13, North, East)
+}

test/dialect/AIE/cascade_configure.mlir

@@ -0,0 +1,20 @@
+//===- cascade_configure.mlir -----------------------------------*- MLIR -*-===//
+//
+// 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 2024 Advanced Micro Devices, Inc.
+//
+//===----------------------------------------------------------------------===//
+
+// RUN: aie-opt %s
+
+module @test {
+  aie.device(xcve2802) {
+    %t13 = aie.tile(1, 3)
+    %t23 = aie.tile(2, 3)
+    aie.configure_cascade(%t13, West, East)
+    aie.configure_cascade(%t23, North, South)
+  }
+}