AIECanonicalizeDevice.cpp

Go to the documentation of this file.

//===- AIECanonicalizeDevice.cpp --------------------------------*- C++ -*-===//
//
// Copyright (C) 2023 Advanced Micro Devices, Inc.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
 
#include "aie/Dialect/AIE/IR/AIEDialect.h"
#include "aie/Dialect/AIE/Transforms/AIEPasses.h"
 
#include "mlir/IR/Builders.h"
#include "mlir/IR/Location.h"
#include "mlir/Pass/Pass.h"
 
namespace xilinx::AIE {
#define GEN_PASS_DEF_AIECANONICALIZEDEVICE
#include "aie/Dialect/AIE/Transforms/AIEPasses.h.inc"
} // namespace xilinx::AIE
 
#define DEBUG_TYPE "aie-canonicalize-device"
 
using namespace mlir;
using namespace xilinx;
using namespace xilinx::AIE;
 
struct AIECanonicalizeDevicePass
    : xilinx::AIE::impl::AIECanonicalizeDeviceBase<AIECanonicalizeDevicePass> {
  void runOnOperation() override {
 
    ModuleOp moduleOp = getOperation();
 
    for (auto deviceOp : moduleOp.getOps<AIETarget>()) {
      (void)deviceOp;
      // We have a device..  Nothing to do.
      return;
    }
 
    // Builder with no insertion point, because we don't want to insert
    // the new op quite yet.
    OpBuilder builder(moduleOp->getContext());
 
    Location location = moduleOp.getLoc();
    auto deviceOp = DeviceOp::create(
        builder, location,
        AIEDeviceAttr::get(builder.getContext(), AIEDevice::xcvc1902),
        StringAttr::get(builder.getContext(),
                        DeviceOp::getDefaultDeviceName()));
 
    deviceOp.getRegion().takeBody(moduleOp.getBodyRegion());
    new (&moduleOp->getRegion(0)) Region(moduleOp);
    moduleOp->getRegion(0).emplaceBlock();
 
    DeviceOp::ensureTerminator(deviceOp.getBodyRegion(), builder, location);
    OpBuilder builder2 = OpBuilder::atBlockBegin(moduleOp.getBody());
    builder2.insert(deviceOp);
  }
};
 
std::unique_ptr<OperationPass<ModuleOp>>
AIE::createAIECanonicalizeDevicePass() {
  return std::make_unique<AIECanonicalizeDevicePass>();
}