Dialects.cpp

Go to the documentation of this file.

//===- Dialects.cpp ---------------------------------------------*- C++ -*-===//
//
// This file is licensed 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
//
// (c) Copyright 2022-2024 Advanced Micro Devices, Inc. or its affiliates
//
//===----------------------------------------------------------------------===//
 
#include <utility>
 
#include "aie-c/Dialects.h"
 
#include "aie/Dialect/AIE/IR/AIEDialect.h"
#include "aie/Dialect/AIEVec/IR/AIEVecDialect.h"
#include "aie/Dialect/AIEX/IR/AIEXDialect.h"
#include "aie/Dialect/XLLVM/XLLVMDialect.h"
 
#include "mlir/CAPI/Registration.h"
 
MLIR_DEFINE_CAPI_DIALECT_REGISTRATION(AIE, aie, xilinx::AIE::AIEDialect)
MLIR_DEFINE_CAPI_DIALECT_REGISTRATION(AIEX, aiex, xilinx::AIEX::AIEXDialect)
MLIR_DEFINE_CAPI_DIALECT_REGISTRATION(AIEVec, aievec,
                                      xilinx::aievec::AIEVecDialect)
MLIR_DEFINE_CAPI_DIALECT_REGISTRATION(XLLVM, xllvm, xilinx::xllvm::XLLVMDialect)
 
//===---------------------------------------------------------------------===//
// ObjectFifoType
//===---------------------------------------------------------------------===//
 
bool aieTypeIsObjectFifoType(MlirType type) {
  return llvm::isa<xilinx::AIE::AIEObjectFifoType>(unwrap(type));
}
 
MlirType aieObjectFifoTypeGet(MlirType type) {
  return wrap(xilinx::AIE::AIEObjectFifoType::get(
      llvm::cast<mlir::MemRefType>(unwrap(type))));
}
 
//===---------------------------------------------------------------------===//
// ObjectFifoSubviewType
//===---------------------------------------------------------------------===//
 
bool aieTypeIsObjectFifoSubviewType(MlirType type) {
  return llvm::isa<xilinx::AIE::AIEObjectFifoSubviewType>(unwrap(type));
}
 
MlirType aieObjectFifoSubviewTypeGet(MlirType type) {
  return wrap(xilinx::AIE::AIEObjectFifoSubviewType::get(
      llvm::cast<mlir::MemRefType>(unwrap(type))));
}
 
//===---------------------------------------------------------------------===//
// BlockFloatType
//===---------------------------------------------------------------------===//
 
bool aieTypeIsBlockFloatType(MlirType type) {
  return llvm::isa<xilinx::AIEX::BlockFloatType>(unwrap(type));
}
 
MlirType aieBlockFloatTypeGet(MlirContext ctx, const std::string &blockType) {
  return wrap(xilinx::AIEX::BlockFloatType::get(unwrap(ctx), blockType));
}
 
//===---------------------------------------------------------------------===//
// TileLike Interface
//===---------------------------------------------------------------------===//
 
bool aieOpImplementsTileLike(MlirOperation op) {
  return llvm::isa<xilinx::AIE::TileLike>(unwrap(op));
}
 
bool aieTileLikeIsCoreTile(MlirOperation op) {
  auto tileLike = llvm::dyn_cast<xilinx::AIE::TileLike>(unwrap(op));
  return tileLike && tileLike.isCoreTile();
}
 
bool aieTileLikeIsMemTile(MlirOperation op) {
  auto tileLike = llvm::dyn_cast<xilinx::AIE::TileLike>(unwrap(op));
  return tileLike && tileLike.isMemTile();
}
 
bool aieTileLikeIsShimNOCTile(MlirOperation op) {
  auto tileLike = llvm::dyn_cast<xilinx::AIE::TileLike>(unwrap(op));
  return tileLike && tileLike.isShimNOCTile();
}
 
bool aieTileLikeIsShimPLTile(MlirOperation op) {
  auto tileLike = llvm::dyn_cast<xilinx::AIE::TileLike>(unwrap(op));
  return tileLike && tileLike.isShimPLTile();
}
 
bool aieTileLikeIsShimNOCorPLTile(MlirOperation op) {
  auto tileLike = llvm::dyn_cast<xilinx::AIE::TileLike>(unwrap(op));
  return tileLike && tileLike.isShimNOCorPLTile();
}