[Tools] Register ArmSVE and ArmSME dialects (#17887)
This allows these dialects to be used with tools like `iree-opt` which
can be helpful for testing and debugging.
Signed-off-by: Benjamin Maxwell <benjamin.maxwell@arm.com>
diff --git a/compiler/src/iree/compiler/Tools/BUILD.bazel b/compiler/src/iree/compiler/Tools/BUILD.bazel
index 95c620c..e57ef6b 100644
--- a/compiler/src/iree/compiler/Tools/BUILD.bazel
+++ b/compiler/src/iree/compiler/Tools/BUILD.bazel
@@ -88,6 +88,8 @@
"@llvm-project//mlir:AffineTransforms",
"@llvm-project//mlir:ArmNeon2dToIntr",
"@llvm-project//mlir:ArmNeonDialect",
+ "@llvm-project//mlir:ArmSVEDialect",
+ "@llvm-project//mlir:ArmSMEDialect",
"@llvm-project//mlir:BufferizationDialect",
"@llvm-project//mlir:ComplexDialect",
"@llvm-project//mlir:ControlFlowDialect",
diff --git a/compiler/src/iree/compiler/Tools/CMakeLists.txt b/compiler/src/iree/compiler/Tools/CMakeLists.txt
index ee8d820..0caa69d 100644
--- a/compiler/src/iree/compiler/Tools/CMakeLists.txt
+++ b/compiler/src/iree/compiler/Tools/CMakeLists.txt
@@ -78,6 +78,8 @@
MLIRAffineDialect
MLIRAffineTransforms
MLIRArmNeonDialect
+ MLIRArmSVEDialect
+ MLIRArmSMEDialect
MLIRArmNeon2dToIntr
MLIRBufferizationDialect
MLIRComplexDialect
diff --git a/compiler/src/iree/compiler/Tools/init_mlir_dialects.h b/compiler/src/iree/compiler/Tools/init_mlir_dialects.h
index e259f83..2148ff2 100644
--- a/compiler/src/iree/compiler/Tools/init_mlir_dialects.h
+++ b/compiler/src/iree/compiler/Tools/init_mlir_dialects.h
@@ -16,6 +16,8 @@
#include "mlir/Dialect/Affine/IR/AffineOps.h"
#include "mlir/Dialect/Affine/TransformOps/AffineTransformOps.h"
#include "mlir/Dialect/ArmNeon/ArmNeonDialect.h"
+#include "mlir/Dialect/ArmSME/IR/ArmSME.h"
+#include "mlir/Dialect/ArmSVE/IR/ArmSVEDialect.h"
#include "mlir/Dialect/Bufferization/IR/Bufferization.h"
#include "mlir/Dialect/Bufferization/TransformOps/BufferizationTransformOps.h"
#include "mlir/Dialect/Complex/IR/Complex.h"
@@ -50,6 +52,7 @@
#include "mlir/Dialect/Transform/LoopExtension/LoopExtension.h"
#include "mlir/Dialect/Vector/IR/VectorOps.h"
#include "mlir/Dialect/Vector/TransformOps/VectorTransformOps.h"
+
#include "mlir/IR/Dialect.h"
#ifdef IREE_HAVE_C_OUTPUT_FORMAT
@@ -80,6 +83,8 @@
quant::QuantizationDialect,
spirv::SPIRVDialect,
arm_neon::ArmNeonDialect,
+ arm_sve::ArmSVEDialect,
+ arm_sme::ArmSMEDialect,
func::FuncDialect,
mlir::arith::ArithDialect,
vector::VectorDialect,
