MLIR  19.0.0git
InitAllDialects.h
Go to the documentation of this file.
1 //===- InitAllDialects.h - MLIR Dialects Registration -----------*- C++ -*-===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // This file defines a helper to trigger the registration of all dialects and
10 // passes to the system.
11 //
12 //===----------------------------------------------------------------------===//
13 
14 #ifndef MLIR_INITALLDIALECTS_H_
15 #define MLIR_INITALLDIALECTS_H_
16 
36 #include "mlir/Dialect/DLTI/DLTI.h"
95 #include "mlir/IR/Dialect.h"
100 
101 namespace mlir {
102 
103 /// Add all the MLIR dialects to the provided registry.
104 inline void registerAllDialects(DialectRegistry &registry) {
105  // clang-format off
106  registry.insert<acc::OpenACCDialect,
107  affine::AffineDialect,
108  amdgpu::AMDGPUDialect,
109  amx::AMXDialect,
110  arith::ArithDialect,
111  arm_neon::ArmNeonDialect,
112  arm_sme::ArmSMEDialect,
113  arm_sve::ArmSVEDialect,
114  async::AsyncDialect,
115  bufferization::BufferizationDialect,
116  cf::ControlFlowDialect,
117  complex::ComplexDialect,
118  DLTIDialect,
119  emitc::EmitCDialect,
120  func::FuncDialect,
121  gpu::GPUDialect,
122  index::IndexDialect,
123  irdl::IRDLDialect,
124  linalg::LinalgDialect,
125  LLVM::LLVMDialect,
126  math::MathDialect,
127  memref::MemRefDialect,
128  mesh::MeshDialect,
129  ml_program::MLProgramDialect,
130  mpi::MPIDialect,
131  nvgpu::NVGPUDialect,
132  NVVM::NVVMDialect,
133  omp::OpenMPDialect,
134  pdl::PDLDialect,
135  pdl_interp::PDLInterpDialect,
136  polynomial::PolynomialDialect,
137  quant::QuantizationDialect,
138  ROCDL::ROCDLDialect,
139  scf::SCFDialect,
140  shape::ShapeDialect,
141  sparse_tensor::SparseTensorDialect,
142  spirv::SPIRVDialect,
143  tensor::TensorDialect,
144  tosa::TosaDialect,
145  transform::TransformDialect,
146  ub::UBDialect,
147  vector::VectorDialect,
148  x86vector::X86VectorDialect,
149  xegpu::XeGPUDialect>();
150  // clang-format on
151 
152  // Register all external models.
159  registry);
190 }
191 
192 /// Append all the MLIR dialects to the registry contained in the given context.
193 inline void registerAllDialects(MLIRContext &context) {
194  DialectRegistry registry;
195  registerAllDialects(registry);
196  context.appendDialectRegistry(registry);
197 }
198 
199 } // namespace mlir
200 
201 #endif // MLIR_INITALLDIALECTS_H_
The DialectRegistry maps a dialect namespace to a constructor for the matching dialect.
MLIRContext is the top-level object for a collection of MLIR operations.
Definition: MLIRContext.h:60
void appendDialectRegistry(const DialectRegistry &registry)
Append the contents of the given dialect registry to the registry associated with this context.
void registerNVVMTargetInterfaceExternalModels(DialectRegistry &registry)
Registers the TargetAttrInterface for the #nvvm.target attribute in the given registry.
Definition: Target.cpp:59
void registerROCDLTargetInterfaceExternalModels(DialectRegistry &registry)
Registers the TargetAttrInterface for the #rocdl.target attribute in the given registry.
Definition: Target.cpp:72
void registerValueBoundsOpInterfaceExternalModels(DialectRegistry &registry)
void registerBufferizableOpInterfaceExternalModels(DialectRegistry &registry)
void registerBufferViewFlowOpInterfaceExternalModels(DialectRegistry &registry)
void registerBufferDeallocationOpInterfaceExternalModels(DialectRegistry &registry)
void registerValueBoundsOpInterfaceExternalModels(DialectRegistry &registry)
void registerBufferizableOpInterfaceExternalModels(DialectRegistry &registry)
void registerCastOpInterfaceExternalModels(DialectRegistry &registry)
void registerBufferizableOpInterfaceExternalModels(DialectRegistry &registry)
void registerBufferDeallocationOpInterfaceExternalModels(DialectRegistry &registry)
void registerBufferDeallocationOpInterfaceExternalModels(DialectRegistry &registry)
void registerAllDialectInterfaceImplementations(DialectRegistry &registry)
void registerRuntimeVerifiableOpInterfaceExternalModels(DialectRegistry &registry)
void registerMemorySlotExternalModels(DialectRegistry &registry)
void registerAllocationOpInterfaceExternalModels(DialectRegistry &registry)
void registerBufferViewFlowOpInterfaceExternalModels(DialectRegistry &registry)
void registerValueBoundsOpInterfaceExternalModels(DialectRegistry &registry)
void registerRuntimeVerifiableOpInterfaceExternalModels(DialectRegistry &registry)
void registerBufferizableOpInterfaceExternalModels(DialectRegistry &registry)
void registerBufferDeallocationOpInterfaceExternalModels(DialectRegistry &registry)
void registerValueBoundsOpInterfaceExternalModels(DialectRegistry &registry)
void registerBufferizableOpInterfaceExternalModels(DialectRegistry &registry)
void registerBufferizableOpInterfaceExternalModels(DialectRegistry &registry)
void registerBufferizableOpInterfaceExternalModels(DialectRegistry &registry)
void registerSPIRVTargetInterfaceExternalModels(DialectRegistry &registry)
Registers the TargetAttrInterface for the #spirv.target_env attribute in the given registry.
Definition: Target.cpp:44
void registerInferTypeOpInterfaceExternalModels(mlir::DialectRegistry &registry)
Registers external models for Infer Type interfaces for tensor ops.
void registerSubsetOpInterfaceExternalModels(DialectRegistry &registry)
void registerFindPayloadReplacementOpInterfaceExternalModels(DialectRegistry &registry)
void registerTilingInterfaceExternalModels(mlir::DialectRegistry &registry)
Registers external models for Tiling interface for tensor ops.
void registerBufferizableOpInterfaceExternalModels(DialectRegistry &registry)
void registerValueBoundsOpInterfaceExternalModels(DialectRegistry &registry)
void registerShardingInterfaceExternalModels(DialectRegistry &registry)
void registerSubsetOpInterfaceExternalModels(DialectRegistry &registry)
void registerBufferizableOpInterfaceExternalModels(DialectRegistry &registry)
void registerValueBoundsOpInterfaceExternalModels(DialectRegistry &registry)
Include the generated interface declarations.
void registerAllDialects(DialectRegistry &registry)
Add all the MLIR dialects to the provided registry.