DialectExtension.cpp

Go to the documentation of this file.

//===- DialectExtension.cpp -------------------------------------*- C++ -*-===//
//
// Copyright (C) 2023-2024 Advanced Micro Devices, Inc.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
 
#include "aie/Dialect/AIEVec/TransformOps/DialectExtension.h"
#include "aie/Dialect/AIEVec/IR/AIEVecDialect.h"
#include "aie/Dialect/AIEVec/TransformOps/AIEVecTransformOps.h"
 
#include "mlir/Dialect/Linalg/IR/Linalg.h"
#include "mlir/Dialect/Transform/IR/TransformDialect.h"
#include "mlir/Dialect/Vector/IR/VectorOps.h"
 
using namespace mlir;
using namespace xilinx;
 
namespace {
 
class AIEVecTransformDialectExtension
    : public transform::TransformDialectExtension<
          AIEVecTransformDialectExtension> {
public:
  MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(AIEVecTransformDialectExtension)
 
  using Base::Base;
 
  void init() {
    declareDependentDialect<linalg::LinalgDialect>();
    declareDependentDialect<vector::VectorDialect>();
 
    registerTransformOps<
#define GET_OP_LIST
#include "aie/Dialect/AIEVec/TransformOps/AIEVecTransformOps.cpp.inc"
        >();
  }
};
 
} // namespace
 
void aievec::registerTransformDialectExtension(DialectRegistry &registry) {
  registry.addExtensions<AIEVecTransformDialectExtension>();
}