AIEXToStandard.cpp

Go to the documentation of this file.

//===- AIEXToStandard.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/AIEX/IR/AIEXDialect.h"
#include "aie/Dialect/AIEX/Transforms/AIEXPasses.h"
 
#include "mlir/IR/IRMapping.h"
#include "mlir/IR/PatternMatch.h"
#include "mlir/Pass/Pass.h"
#include "mlir/Tools/mlir-translate/MlirTranslateMain.h"
#include "mlir/Transforms/DialectConversion.h"
 
namespace xilinx::AIEX {
#define GEN_PASS_DEF_AIEXTOSTANDARD
#include "aie/Dialect/AIEX/Transforms/AIEXPasses.h.inc"
} // namespace xilinx::AIEX
 
using namespace mlir;
using namespace xilinx;
using namespace xilinx::AIE;
using namespace xilinx::AIEX;
 
template <typename MyAIEXOp>
struct AIEXOpRemoval : OpConversionPattern<MyAIEXOp> {
  using OpConversionPattern<MyAIEXOp>::OpConversionPattern;
  using OpAdaptor = typename MyAIEXOp::Adaptor;
  ModuleOp &module;
 
  AIEXOpRemoval(MLIRContext *context, ModuleOp &m, PatternBenefit benefit = 1)
      : OpConversionPattern<MyAIEXOp>(context, benefit), module(m) {}
 
  LogicalResult
  matchAndRewrite(MyAIEXOp op, OpAdaptor adaptor,
                  ConversionPatternRewriter &rewriter) const override {
    Operation *Op = op.getOperation();
    rewriter.eraseOp(Op);
    return success();
  }
};
 
struct AIEXToStandardPass
    : xilinx::AIEX::impl::AIEXToStandardBase<AIEXToStandardPass> {
  void runOnOperation() override {
 
    ModuleOp m = getOperation();
    ConversionTarget target(getContext());
    RewritePatternSet removepatterns(&getContext());
    removepatterns.add<AIEXOpRemoval<NpuDmaMemcpyNdOp>>(m.getContext(), m);
    removepatterns.add<AIEXOpRemoval<NpuDmaWaitOp>>(m.getContext(), m);
    removepatterns.add<AIEXOpRemoval<NpuPushQueueOp>>(m.getContext(), m);
    removepatterns.add<AIEXOpRemoval<NpuWriteRTPOp>>(m.getContext(), m);
    removepatterns.add<AIEXOpRemoval<NpuWrite32Op>>(m.getContext(), m);
    removepatterns.add<AIEXOpRemoval<NpuSyncOp>>(m.getContext(), m);
    removepatterns.add<AIEXOpRemoval<NpuWriteBdOp>>(m.getContext(), m);
    removepatterns.add<AIEXOpRemoval<NpuAddressPatchOp>>(m.getContext(), m);
    removepatterns.add<AIEXOpRemoval<NpuPreemptOp>>(m.getContext(), m);
    removepatterns.add<AIEXOpRemoval<NpuCreateScratchpadOp>>(m.getContext(), m);
    removepatterns.add<AIEXOpRemoval<NpuUpdateFromScratchpadOp>>(m.getContext(),
                                                                 m);
    removepatterns.add<AIEXOpRemoval<ScratchpadParameterOp>>(m.getContext(), m);
    removepatterns.add<AIEXOpRemoval<SyncScratchpadParametersFromHostOp>>(
        m.getContext(), m);
 
    if (failed(applyPartialConversion(m, target, std::move(removepatterns))))
      signalPassFailure();
  }
};
 
std::unique_ptr<OperationPass<ModuleOp>> AIEX::createAIEXToStandardPass() {
  return std::make_unique<AIEXToStandardPass>();
}