AIELowerMulticast.cpp

Go to the documentation of this file.

//===- AIELowerMulticast.cpp ------------------------------------*- C++ -*-===//
//
// Copyright (C) 2020-2022 Xilinx, Inc.
// Copyright (C) 2022-2026 Advanced Micro Devices, Inc.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
 
#include "aie/Dialect/AIE/IR/AIEDialect.h"
#include "aie/Dialect/AIEX/IR/AIEXDialect.h"
#include "aie/Dialect/AIEX/Transforms/AIEXPasses.h"
 
#include "mlir/IR/Attributes.h"
#include "mlir/IR/PatternMatch.h"
#include "mlir/Pass/Pass.h"
#include "mlir/Tools/mlir-translate/MlirTranslateMain.h"
#include "mlir/Transforms/DialectConversion.h"
 
#include "llvm/ADT/Twine.h"
 
namespace xilinx::AIEX {
#define GEN_PASS_DEF_AIEMULTICAST
#include "aie/Dialect/AIEX/Transforms/AIEXPasses.h.inc"
} // namespace xilinx::AIEX
 
#define DEBUG_TYPE "aie-lower-multicast"
 
using namespace mlir;
using namespace xilinx;
using namespace xilinx::AIE;
using namespace xilinx::AIEX;
 
template <typename MyOp>
struct AIEOpRemoval : public OpConversionPattern<MyOp> {
  using OpConversionPattern<MyOp>::OpConversionPattern;
  using OpAdaptor = typename MyOp::Adaptor;
 
  AIEOpRemoval(MLIRContext *context, PatternBenefit benefit = 1)
      : OpConversionPattern<MyOp>(context, benefit) {}
 
  LogicalResult
  matchAndRewrite(MyOp op, OpAdaptor adaptor,
                  ConversionPatternRewriter &rewriter) const override {
    Operation *Op = op.getOperation();
 
    rewriter.eraseOp(Op);
    return success();
  }
};
 
struct AIELowerMulticastPass
    : public xilinx::AIEX::impl::AIEMulticastBase<AIELowerMulticastPass> {
  void runOnOperation() override {
 
    DeviceOp device = getOperation();
    OpBuilder builder = OpBuilder::atBlockTerminator(device.getBody());
 
    for (auto multicast : device.getOps<MulticastOp>()) {
      Region &r = multicast.getPorts();
      Block &b = r.front();
      Port sourcePort = multicast.port();
      TileOp srcTile = cast<TileOp>(multicast.getTile().getDefiningOp());
      for (Operation &Op : b.getOperations()) {
        if (MultiDestOp multiDest = dyn_cast<MultiDestOp>(Op)) {
          TileOp destTile = cast<TileOp>(multiDest.getTile().getDefiningOp());
          Port destPort = multiDest.port();
          FlowOp::create(builder, multiDest.getLoc(), srcTile,
                         sourcePort.bundle, sourcePort.channel, destTile,
                         destPort.bundle, destPort.channel);
        }
      }
    }
 
    ConversionTarget target(getContext());
    RewritePatternSet patterns(&getContext());
 
    patterns.add<AIEOpRemoval<MulticastOp>>(device.getContext());
 
    if (failed(applyPartialConversion(device, target, std::move(patterns))))
      signalPassFailure();
  }
};
 
std::unique_ptr<OperationPass<DeviceOp>>
xilinx::AIEX::createAIELowerMulticastPass() {
  return std::make_unique<AIELowerMulticastPass>();
}