hanchenye-scalehls/include/scalehls/InitAllDialects.h

48 lines
1.4 KiB
C++

//===----------------------------------------------------------------------===//
//
// Copyright 2020-2021 The ScaleHLS Authors.
//
//===----------------------------------------------------------------------===//
#ifndef SCALEHLS_INITALLDIALECTS_H
#define SCALEHLS_INITALLDIALECTS_H
#include "mlir/Dialect/Bufferization/IR/Bufferization.h"
#include "mlir/Dialect/DLTI/DLTI.h"
#include "mlir/Dialect/Func/IR/FuncOps.h"
#include "mlir/Dialect/LLVMIR/LLVMDialect.h"
#include "mlir/Dialect/Linalg/IR/Linalg.h"
#include "mlir/Dialect/SCF/SCF.h"
#include "mlir/Dialect/Tosa/IR/TosaOps.h"
#include "mlir/Dialect/Vector/IR/VectorOps.h"
#include "scalehls/Dialect/HLS/HLS.h"
namespace mlir {
namespace scalehls {
// Add all the ScaleHLS dialects to the provided registry.
inline void registerAllDialects(mlir::DialectRegistry &registry) {
// clang-format off
registry.insert<
mlir::func::FuncDialect,
mlir::tosa::TosaDialect,
mlir::linalg::LinalgDialect,
mlir::memref::MemRefDialect,
mlir::bufferization::BufferizationDialect,
mlir::AffineDialect,
mlir::math::MathDialect,
mlir::arith::ArithmeticDialect,
mlir::vector::VectorDialect,
mlir::scf::SCFDialect,
mlir::scalehls::hls::HLSDialect,
mlir::LLVM::LLVMDialect,
mlir::DLTIDialect
>();
// clang-format on
}
} // namespace scalehls
} // namespace mlir
#endif // SCALEHLS_INITALLDIALECTS_H