mirror of
https://github.com/intel/llvm.git
synced 2026-01-14 11:57:39 +08:00
d8b84be107
Introduces a Transform-dialect SMT-extension so that we can have an op to express constrains on Transform-dialect params, in particular when these params are knobs -- see transform.tune.knob -- and can hence be seen as symbolic variables. This op allows expressing joint constraints over multiple params/knobs together. While the op's semantics are clearly defined, per SMTLIB, the interpreted semantics -- i.e. the `apply()` method -- for now just defaults to failure. In the future we should support attaching an implementation so that users can Bring Your Own Solver and thereby control performance of interpreting the op. For now the main usage is to walk schedule IR and collect these constraints so that knobs can be rewritten to constants that satisfy the constraints.
121 lines
6.1 KiB
C++
121 lines
6.1 KiB
C++
//===- RegisterAllExtensions.cpp - MLIR Extension Registration --*- C++ -*-===//
|
|
//
|
|
// Part of the LLVM Project, 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file defines a helper to trigger the registration of all dialect
|
|
// extensions to the system.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/InitAllExtensions.h"
|
|
|
|
#include "mlir/Conversion/ArithToEmitC/ArithToEmitC.h"
|
|
#include "mlir/Conversion/ArithToLLVM/ArithToLLVM.h"
|
|
#include "mlir/Conversion/ComplexToLLVM/ComplexToLLVM.h"
|
|
#include "mlir/Conversion/ControlFlowToLLVM/ControlFlowToLLVM.h"
|
|
#include "mlir/Conversion/FuncToEmitC/FuncToEmitC.h"
|
|
#include "mlir/Conversion/FuncToLLVM/ConvertFuncToLLVM.h"
|
|
#include "mlir/Conversion/GPUCommon/GPUToLLVM.h"
|
|
#include "mlir/Conversion/GPUToNVVM/GPUToNVVM.h"
|
|
#include "mlir/Conversion/IndexToLLVM/IndexToLLVM.h"
|
|
#include "mlir/Conversion/MPIToLLVM/MPIToLLVM.h"
|
|
#include "mlir/Conversion/MathToLLVM/MathToLLVM.h"
|
|
#include "mlir/Conversion/MemRefToEmitC/MemRefToEmitC.h"
|
|
#include "mlir/Conversion/MemRefToLLVM/MemRefToLLVM.h"
|
|
#include "mlir/Conversion/NVVMToLLVM/NVVMToLLVM.h"
|
|
#include "mlir/Conversion/OpenMPToLLVM/ConvertOpenMPToLLVM.h"
|
|
#include "mlir/Conversion/PtrToLLVM/PtrToLLVM.h"
|
|
#include "mlir/Conversion/SCFToEmitC/SCFToEmitC.h"
|
|
#include "mlir/Conversion/UBToLLVM/UBToLLVM.h"
|
|
#include "mlir/Conversion/VectorToLLVM/ConvertVectorToLLVM.h"
|
|
#include "mlir/Conversion/XeVMToLLVM/XeVMToLLVM.h"
|
|
#include "mlir/Dialect/AMX/Transforms.h"
|
|
#include "mlir/Dialect/Affine/TransformOps/AffineTransformOps.h"
|
|
#include "mlir/Dialect/ArmNeon/TransformOps/ArmNeonVectorTransformOps.h"
|
|
#include "mlir/Dialect/ArmSVE/TransformOps/ArmSVEVectorTransformOps.h"
|
|
#include "mlir/Dialect/Bufferization/TransformOps/BufferizationTransformOps.h"
|
|
#include "mlir/Dialect/DLTI/TransformOps/DLTITransformOps.h"
|
|
#include "mlir/Dialect/Func/Extensions/AllExtensions.h"
|
|
#include "mlir/Dialect/Func/TransformOps/FuncTransformOps.h"
|
|
#include "mlir/Dialect/GPU/TransformOps/GPUTransformOps.h"
|
|
#include "mlir/Dialect/Linalg/TransformOps/DialectExtension.h"
|
|
#include "mlir/Dialect/MemRef/TransformOps/MemRefTransformOps.h"
|
|
#include "mlir/Dialect/NVGPU/TransformOps/NVGPUTransformOps.h"
|
|
#include "mlir/Dialect/SCF/TransformOps/SCFTransformOps.h"
|
|
#include "mlir/Dialect/SparseTensor/TransformOps/SparseTensorTransformOps.h"
|
|
#include "mlir/Dialect/Tensor/Extensions/AllExtensions.h"
|
|
#include "mlir/Dialect/Tensor/TransformOps/TensorTransformOps.h"
|
|
#include "mlir/Dialect/Transform/DebugExtension/DebugExtension.h"
|
|
#include "mlir/Dialect/Transform/IRDLExtension/IRDLExtension.h"
|
|
#include "mlir/Dialect/Transform/LoopExtension/LoopExtension.h"
|
|
#include "mlir/Dialect/Transform/PDLExtension/PDLExtension.h"
|
|
#include "mlir/Dialect/Transform/SMTExtension/SMTExtension.h"
|
|
#include "mlir/Dialect/Transform/TuneExtension/TuneExtension.h"
|
|
#include "mlir/Dialect/Vector/TransformOps/VectorTransformOps.h"
|
|
#include "mlir/Target/LLVMIR/Dialect/Builtin/BuiltinToLLVMIRTranslation.h"
|
|
#include "mlir/Target/LLVMIR/Dialect/GPU/GPUToLLVMIRTranslation.h"
|
|
#include "mlir/Target/LLVMIR/Dialect/LLVMIR/LLVMToLLVMIRTranslation.h"
|
|
#include "mlir/Target/LLVMIR/Dialect/ROCDL/ROCDLToLLVMIRTranslation.h"
|
|
#include "mlir/Target/LLVMIR/Dialect/XeVM/XeVMToLLVMIRTranslation.h"
|
|
|
|
/// This function may be called to register all MLIR dialect extensions with the
|
|
/// provided registry.
|
|
/// If you're building a compiler, you generally shouldn't use this: you would
|
|
/// individually register the specific extensions that are useful for the
|
|
/// pipelines and transformations you are using.
|
|
void mlir::registerAllExtensions(DialectRegistry ®istry) {
|
|
// Register all conversions to LLVM extensions.
|
|
registerConvertArithToEmitCInterface(registry);
|
|
arith::registerConvertArithToLLVMInterface(registry);
|
|
registerConvertComplexToLLVMInterface(registry);
|
|
cf::registerConvertControlFlowToLLVMInterface(registry);
|
|
func::registerAllExtensions(registry);
|
|
tensor::registerAllExtensions(registry);
|
|
registerConvertFuncToEmitCInterface(registry);
|
|
registerConvertFuncToLLVMInterface(registry);
|
|
index::registerConvertIndexToLLVMInterface(registry);
|
|
registerConvertMathToLLVMInterface(registry);
|
|
mpi::registerConvertMPIToLLVMInterface(registry);
|
|
registerConvertMemRefToEmitCInterface(registry);
|
|
registerConvertMemRefToLLVMInterface(registry);
|
|
registerConvertNVVMToLLVMInterface(registry);
|
|
ptr::registerConvertPtrToLLVMInterface(registry);
|
|
registerConvertOpenMPToLLVMInterface(registry);
|
|
registerConvertSCFToEmitCInterface(registry);
|
|
ub::registerConvertUBToLLVMInterface(registry);
|
|
registerConvertAMXToLLVMInterface(registry);
|
|
gpu::registerConvertGpuToLLVMInterface(registry);
|
|
NVVM::registerConvertGpuToNVVMInterface(registry);
|
|
vector::registerConvertVectorToLLVMInterface(registry);
|
|
registerConvertXeVMToLLVMInterface(registry);
|
|
|
|
// Register all transform dialect extensions.
|
|
affine::registerTransformDialectExtension(registry);
|
|
bufferization::registerTransformDialectExtension(registry);
|
|
dlti::registerTransformDialectExtension(registry);
|
|
func::registerTransformDialectExtension(registry);
|
|
gpu::registerTransformDialectExtension(registry);
|
|
linalg::registerTransformDialectExtension(registry);
|
|
memref::registerTransformDialectExtension(registry);
|
|
nvgpu::registerTransformDialectExtension(registry);
|
|
scf::registerTransformDialectExtension(registry);
|
|
sparse_tensor::registerTransformDialectExtension(registry);
|
|
tensor::registerTransformDialectExtension(registry);
|
|
transform::registerDebugExtension(registry);
|
|
transform::registerIRDLExtension(registry);
|
|
transform::registerLoopExtension(registry);
|
|
transform::registerPDLExtension(registry);
|
|
transform::registerSMTExtension(registry);
|
|
transform::registerTuneExtension(registry);
|
|
vector::registerTransformDialectExtension(registry);
|
|
arm_neon::registerTransformDialectExtension(registry);
|
|
arm_sve::registerTransformDialectExtension(registry);
|
|
|
|
// Translation extensions need to be registered by calling
|
|
// `registerAllToLLVMIRTranslations` (see All.h).
|
|
}
|