KanagawaContainersToHW.cpp

Go to the documentation of this file.

//===- KanagawaContainersToHW.cpp -----------------------------------------===//
//
// Part of the LLVM Project, 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
//
//===----------------------------------------------------------------------===//
 
#include "circt/Dialect/Kanagawa/KanagawaOps.h"
#include "circt/Dialect/Kanagawa/KanagawaPasses.h"
#include "mlir/Pass/Pass.h"
 
#include "circt/Dialect/HW/HWOps.h"
#include "circt/Dialect/Kanagawa/KanagawaDialect.h"
#include "circt/Dialect/Kanagawa/KanagawaOps.h"
#include "circt/Dialect/Kanagawa/KanagawaPasses.h"
#include "circt/Dialect/Kanagawa/KanagawaTypes.h"
 
#include "circt/Support/Namespace.h"
#include "mlir/IR/OperationSupport.h"
#include "mlir/Transforms/DialectConversion.h"
#include "llvm/ADT/TypeSwitch.h"
 
namespace circt {
namespace kanagawa {
#define GEN_PASS_DEF_KANAGAWACONTAINERSTOHW
#include "circt/Dialect/Kanagawa/KanagawaPasses.h.inc"
} // namespace kanagawa
} // namespace circt
 
using namespace circt;
using namespace kanagawa;
 
namespace {
 
// Analysis result for generating the port interface of a container + a bit of
// port op caching.
struct ContainerPortInfo {
  std::unique_ptr<hw::ModulePortInfo> hwPorts;
 
  // A mapping between the port name and the port op within the container.
  llvm::DenseMap<StringAttr, InputPortOp> opInputs;
 
  // A mapping between the port name and the port op within the container.
  llvm::DenseMap<StringAttr, OutputPortOp> opOutputs;
 
  // A mapping between port symbols and their corresponding port name.
  llvm::DenseMap<StringAttr, StringAttr> portSymbolsToPortName;
 
  ContainerPortInfo() = default;
  ContainerPortInfo(ContainerOp container) {
    SmallVector<hw::PortInfo, 4> inputs, outputs;
    auto *ctx = container.getContext();
 
    // Copies all attributes from a port, except for the port symbol, name, and
    // type.
    auto copyPortAttrs = [ctx](auto port) {
      llvm::DenseSet<StringAttr> elidedAttrs;
      elidedAttrs.insert(port.getInnerSymAttrName());
      elidedAttrs.insert(port.getTypeAttrName());
      elidedAttrs.insert(port.getNameAttrName());
      llvm::SmallVector<NamedAttribute> attrs;
      for (NamedAttribute namedAttr : port->getAttrs()) {
        if (elidedAttrs.contains(namedAttr.getName()))
          continue;
        attrs.push_back(namedAttr);
      }
      return DictionaryAttr::get(ctx, attrs);
    };
 
    // Gather in and output port ops to define the hw.module interface. Here, we
    // also perform uniqueing of the port names.
    Namespace portNs;
    for (auto input : container.getBodyBlock()->getOps<InputPortOp>()) {
      auto uniquePortName =
          StringAttr::get(ctx, portNs.newName(input.getNameHint()));
      opInputs[uniquePortName] = input;
      hw::PortInfo portInfo;
      portInfo.name = uniquePortName;
      portSymbolsToPortName[input.getInnerSym().getSymName()] = uniquePortName;
      portInfo.type = cast<PortOpInterface>(input.getOperation()).getPortType();
      portInfo.dir = hw::ModulePort::Direction::Input;
      portInfo.attrs = copyPortAttrs(input);
      inputs.push_back(portInfo);
    }
 
    for (auto output : container.getBodyBlock()->getOps<OutputPortOp>()) {
      auto uniquePortName =
          StringAttr::get(ctx, portNs.newName(output.getNameAttr().getValue()));
      opOutputs[uniquePortName] = output;
 
      hw::PortInfo portInfo;
      portInfo.name = uniquePortName;
      portSymbolsToPortName[output.getInnerSym().getSymName()] = uniquePortName;
      portInfo.type =
          cast<PortOpInterface>(output.getOperation()).getPortType();
      portInfo.dir = hw::ModulePort::Direction::Output;
      portInfo.attrs = copyPortAttrs(output);
      outputs.push_back(portInfo);
    }
    hwPorts = std::make_unique<hw::ModulePortInfo>(inputs, outputs);
  }
};
 
using ContainerPortInfoMap =
    llvm::DenseMap<hw::InnerRefAttr, ContainerPortInfo>;
using ContainerHWModSymbolMap = llvm::DenseMap<hw::InnerRefAttr, StringAttr>;
 
static StringAttr concatNames(mlir::StringAttr lhs, mlir::StringAttr rhs) {
  return StringAttr::get(lhs.getContext(), lhs.strref() + "_" + rhs.strref());
}
 
struct ContainerOpConversionPattern : public OpConversionPattern<ContainerOp> {
  ContainerOpConversionPattern(MLIRContext *ctx, Namespace &modNamespace,
                               ContainerPortInfoMap &portOrder,
                               ContainerHWModSymbolMap &modSymMap)
      : OpConversionPattern<ContainerOp>(ctx), modNamespace(modNamespace),
        portOrder(portOrder), modSymMap(modSymMap) {}
 
  LogicalResult
  matchAndRewrite(ContainerOp op, OpAdaptor adaptor,
                  ConversionPatternRewriter &rewriter) const override {
    auto design = op->getParentOfType<DesignOp>();
    rewriter.setInsertionPoint(design);
 
    // Generate and de-alias the hw.module name.
    // If the container is a top level container, ignore the design name.
    StringAttr hwmodName;
    if (op.getIsTopLevel())
      hwmodName = op.getNameHintAttr();
    else
      hwmodName =
          concatNames(op.getInnerRef().getModule(), op.getNameHintAttr());
 
    hwmodName = StringAttr::get(op.getContext(),
                                modNamespace.newName(hwmodName.getValue()));
 
    const ContainerPortInfo &cpi = portOrder.at(op.getInnerRef());
    auto hwMod =
        rewriter.create<hw::HWModuleOp>(op.getLoc(), hwmodName, *cpi.hwPorts);
    modSymMap[op.getInnerRef()] = hwMod.getSymNameAttr();
 
    hw::OutputOp outputOp =
        cast<hw::OutputOp>(hwMod.getBodyBlock()->getTerminator());
 
    // Replace all of the reads of the inputs to use the input block arguments.
    for (auto [idx, input] : llvm::enumerate(cpi.hwPorts->getInputs())) {
      Value barg = hwMod.getBodyBlock()->getArgument(idx);
      InputPortOp inputPort = cpi.opInputs.at(input.name);
      // Replace all reads of the input port with the input block argument.
      for (auto *user : inputPort.getOperation()->getUsers()) {
        auto reader = dyn_cast<PortReadOp>(user);
        if (!reader)
          return rewriter.notifyMatchFailure(
              user, "expected only kanagawa.port.read ops of the input port");
 
        rewriter.replaceOp(reader, barg);
      }
 
      rewriter.eraseOp(inputPort);
    }
 
    // Adjust the hw.output op to use kanagawa.port.write values
    llvm::SmallVector<Value> outputValues;
    for (auto [idx, output] : llvm::enumerate(cpi.hwPorts->getOutputs())) {
      auto outputPort = cpi.opOutputs.at(output.name);
      // Locate the write to the output op.
      auto users = outputPort->getUsers();
      size_t nUsers = std::distance(users.begin(), users.end());
      if (nUsers != 1)
        return outputPort->emitOpError()
               << "expected exactly one kanagawa.port.write op of the output "
                  "port: "
               << output.name.str() << " found: " << nUsers;
      auto writer = cast<PortWriteOp>(*users.begin());
      outputValues.push_back(writer.getValue());
      rewriter.eraseOp(outputPort);
      rewriter.eraseOp(writer);
    }
 
    rewriter.mergeBlocks(&op.getBodyRegion().front(), hwMod.getBodyBlock());
 
    // Rewrite the hw.output op.
    rewriter.eraseOp(outputOp);
    rewriter.setInsertionPointToEnd(hwMod.getBodyBlock());
    outputOp = rewriter.create<hw::OutputOp>(op.getLoc(), outputValues);
    rewriter.eraseOp(op);
    return success();
  }
 
  Namespace &modNamespace;
  ContainerPortInfoMap &portOrder;
  ContainerHWModSymbolMap &modSymMap;
};
 
struct ContainerInstanceOpConversionPattern
    : public OpConversionPattern<ContainerInstanceOp> {
 
  ContainerInstanceOpConversionPattern(MLIRContext *ctx,
                                       ContainerPortInfoMap &portOrder,
                                       ContainerHWModSymbolMap &modSymMap)
      : OpConversionPattern<ContainerInstanceOp>(ctx), portOrder(portOrder),
        modSymMap(modSymMap) {}
 
  LogicalResult
  matchAndRewrite(ContainerInstanceOp op, OpAdaptor adaptor,
                  ConversionPatternRewriter &rewriter) const override {
    rewriter.setInsertionPoint(op);
    llvm::SmallVector<Value> operands;
 
    const ContainerPortInfo &cpi =
        portOrder.at(op.getResult().getType().getScopeRef());
 
    // Gather the get_port ops that target the instance
    llvm::DenseMap<StringAttr, PortReadOp> outputReadsToReplace;
    llvm::DenseMap<StringAttr, PortWriteOp> inputWritesToUse;
    llvm::SmallVector<Operation *> getPortsToErase;
    for (auto *user : op->getUsers()) {
      auto getPort = dyn_cast<GetPortOp>(user);
      if (!getPort)
        return rewriter.notifyMatchFailure(
            user, "expected only kanagawa.get_port op usage of the instance");
 
      for (auto *user : getPort->getUsers()) {
        auto res =
            llvm::TypeSwitch<Operation *, LogicalResult>(user)
                .Case<PortReadOp>([&](auto read) {
                  auto [it, inserted] = outputReadsToReplace.insert(
                      {cpi.portSymbolsToPortName.at(
                           getPort.getPortSymbolAttr().getAttr()),
                       read});
                  if (!inserted)
                    return rewriter.notifyMatchFailure(
                        read, "expected only one kanagawa.port.read op of the "
                              "output port");
                  return success();
                })
                .Case<PortWriteOp>([&](auto write) {
                  auto [it, inserted] = inputWritesToUse.insert(
                      {cpi.portSymbolsToPortName.at(
                           getPort.getPortSymbolAttr().getAttr()),
                       write});
                  if (!inserted)
                    return rewriter.notifyMatchFailure(
                        write,
                        "expected only one kanagawa.port.write op of the input "
                        "port");
                  return success();
                })
                .Default([&](auto op) {
                  return rewriter.notifyMatchFailure(
                      op, "expected only kanagawa.port.read or "
                          "kanagawa.port.write ops "
                          "of the "
                          "instance");
                });
        if (failed(res))
          return failure();
      }
      getPortsToErase.push_back(getPort);
    }
 
    // Grab the operands in the order of the hw.module ports.
    size_t nInputPorts = std::distance(cpi.hwPorts->getInputs().begin(),
                                       cpi.hwPorts->getInputs().end());
    if (nInputPorts != inputWritesToUse.size()) {
      std::string errMsg;
      llvm::raw_string_ostream ers(errMsg);
      ers << "Error when lowering instance ";
      op.print(ers, mlir::OpPrintingFlags().printGenericOpForm());
 
      ers << "\nexpected exactly one kanagawa.port.write op of each input "
             "port. "
             "Mising port assignments were:\n";
      for (auto input : cpi.hwPorts->getInputs()) {
        if (inputWritesToUse.find(input.name) == inputWritesToUse.end())
          ers << "\t" << input.name << "\n";
      }
      return rewriter.notifyMatchFailure(op, errMsg);
    }
    for (auto input : cpi.hwPorts->getInputs()) {
      auto writeOp = inputWritesToUse.at(input.name);
      operands.push_back(writeOp.getValue());
      rewriter.eraseOp(writeOp);
    }
 
    // Determine the result types.
    llvm::SmallVector<Type> retTypes;
    for (auto output : cpi.hwPorts->getOutputs())
      retTypes.push_back(output.type);
 
    // Gather arg and res names
    // TODO: @mortbopet - this should be part of ModulePortInfo
    llvm::SmallVector<Attribute> argNames, resNames;
    llvm::transform(cpi.hwPorts->getInputs(), std::back_inserter(argNames),
                    [](auto port) { return port.name; });
    llvm::transform(cpi.hwPorts->getOutputs(), std::back_inserter(resNames),
                    [](auto port) { return port.name; });
 
    // Create the hw.instance op.
    StringRef moduleName = modSymMap[op.getTargetNameAttr()];
    auto hwInst = rewriter.create<hw::InstanceOp>(
        op.getLoc(), retTypes, op.getInnerSym().getSymName(), moduleName,
        operands, rewriter.getArrayAttr(argNames),
        rewriter.getArrayAttr(resNames),
        /*parameters*/ rewriter.getArrayAttr({}), /*innerSym*/ nullptr);
 
    // Replace the reads of the output ports with the hw.instance results.
    for (auto [output, value] :
         llvm::zip(cpi.hwPorts->getOutputs(), hwInst.getResults())) {
      auto outputReadIt = outputReadsToReplace.find(output.name);
      if (outputReadIt == outputReadsToReplace.end())
        continue;
      // TODO: RewriterBase::replaceAllUsesWith is not currently supported by
      // DialectConversion. Using it may lead to assertions about mutating
      // replaced/erased ops. For now, do this RAUW directly, until
      // ConversionPatternRewriter properly supports RAUW.
      // See https://github.com/llvm/circt/issues/6795.
      outputReadIt->second.getResult().replaceAllUsesWith(value);
      rewriter.eraseOp(outputReadIt->second);
    }
 
    // Erase the get_port ops.
    for (auto *getPort : getPortsToErase)
      rewriter.eraseOp(getPort);
 
    // And finally erase the instance op.
    rewriter.eraseOp(op);
    return success();
  }
 
  ContainerPortInfoMap &portOrder;
  ContainerHWModSymbolMap &modSymMap;
}; // namespace
 
struct ContainersToHWPass
    : public circt::kanagawa::impl::KanagawaContainersToHWBase<
          ContainersToHWPass> {
  void runOnOperation() override;
};
} // anonymous namespace
 
void ContainersToHWPass::runOnOperation() {
  auto *ctx = &getContext();
 
  // Generate module signatures.
  ContainerPortInfoMap portOrder;
  for (auto design : getOperation().getOps<DesignOp>())
    for (auto container : design.getOps<ContainerOp>())
      portOrder.try_emplace(container.getInnerRef(),
                            ContainerPortInfo(container));
 
  ConversionTarget target(*ctx);
  ContainerHWModSymbolMap modSymMap;
  SymbolCache modSymCache;
  modSymCache.addDefinitions(getOperation());
  Namespace modNamespace;
  modNamespace.add(modSymCache);
  target.addIllegalOp<ContainerOp, ContainerInstanceOp>();
  target.markUnknownOpDynamicallyLegal([](Operation *) { return true; });
 
  // Remove the name of the kanagawa.design's from the namespace - The
  // kanagawa.design op will be removed after this pass, and there may be
  // kanagawa.component's inside the design that have the same name as the
  // design; we want that name to persist, and not be falsely considered a
  // duplicate.
  for (auto designOp : getOperation().getOps<DesignOp>())
    modNamespace.erase(designOp.getSymName());
 
  // Parts of the conversion patterns will update operations in place, which in
  // turn requires the updated operations to be legalizeable. These in-place ops
  // also include kanagawa ops that eventually will get replaced once all of the
  // patterns apply.
  target.addLegalDialect<KanagawaDialect>();
 
  RewritePatternSet patterns(ctx);
  patterns.add<ContainerOpConversionPattern>(ctx, modNamespace, portOrder,
                                             modSymMap);
  patterns.add<ContainerInstanceOpConversionPattern>(ctx, portOrder, modSymMap);
 
  if (failed(
          applyPartialConversion(getOperation(), target, std::move(patterns))))
    signalPassFailure();
 
  // Delete empty design ops.
  for (auto design :
       llvm::make_early_inc_range(getOperation().getOps<DesignOp>()))
    if (design.getBody().front().empty())
      design.erase();
}
 
std::unique_ptr<Pass> circt::kanagawa::createContainersToHWPass() {
  return std::make_unique<ContainersToHWPass>();
}