[Mlir-commits] [mlir] 06881d2 - [mlir][ArmSME] Add constructor for `-convert-vector-to-arm-sme` pass (NFC) (#71705)

[llvmlistbot at llvm.org]

Author: Benjamin Maxwell
Date: 2023-11-09T10:03:33Z
New Revision: 06881d222dd7dbea5d1c3bf6fc4527d60e74cb00

URL: https://github.com/llvm/llvm-project/commit/06881d222dd7dbea5d1c3bf6fc4527d60e74cb00
DIFF: https://github.com/llvm/llvm-project/commit/06881d222dd7dbea5d1c3bf6fc4527d60e74cb00.diff

LOG: [mlir][ArmSME] Add constructor for `-convert-vector-to-arm-sme` pass (NFC) (#71705)

This will be needed to construct the pass downstream (i.e. in IREE).

Added: 
    

Modified: 
    mlir/include/mlir/Conversion/Passes.td
    mlir/include/mlir/Conversion/VectorToArmSME/VectorToArmSME.h
    mlir/lib/Conversion/VectorToArmSME/VectorToArmSMEPass.cpp

Removed: 
    


################################################################################
diff  --git a/mlir/include/mlir/Conversion/Passes.td b/mlir/include/mlir/Conversion/Passes.td
index 8b64f6dbe741b93..fce168d6ce09f47 100644
--- a/mlir/include/mlir/Conversion/Passes.td
+++ b/mlir/include/mlir/Conversion/Passes.td
@@ -1204,6 +1204,7 @@ def ConvertVectorToSCF : Pass<"convert-vector-to-scf"> {
 def ConvertVectorToArmSME : Pass<"convert-vector-to-arm-sme"> {
   let summary = "Lower the operations from the vector dialect into the ArmSME "
                 "dialect";
+  let constructor = "mlir::createConvertVectorToArmSMEPass()";
   let description = [{
     Pass that converts vector dialect operations into equivalent ArmSME dialect
     operations.

diff  --git a/mlir/include/mlir/Conversion/VectorToArmSME/VectorToArmSME.h b/mlir/include/mlir/Conversion/VectorToArmSME/VectorToArmSME.h
index 2108e485dae7ba7..2332311ee72cdff 100644
--- a/mlir/include/mlir/Conversion/VectorToArmSME/VectorToArmSME.h
+++ b/mlir/include/mlir/Conversion/VectorToArmSME/VectorToArmSME.h
@@ -21,6 +21,9 @@ class Pass;
 void populateVectorToArmSMEPatterns(RewritePatternSet &patterns,
                                     MLIRContext &ctx);
 
+/// Create a pass to lower operations from the vector dialect to Arm SME.
+std::unique_ptr<Pass> createConvertVectorToArmSMEPass();
+
 } // namespace mlir
 
 #endif // MLIR_CONVERSION_VECTORTOARMSME_VECTORTOARMSME_H_

diff  --git a/mlir/lib/Conversion/VectorToArmSME/VectorToArmSMEPass.cpp b/mlir/lib/Conversion/VectorToArmSME/VectorToArmSMEPass.cpp
index 92025e9fbe82d3e..2601f31be11a32c 100644
--- a/mlir/lib/Conversion/VectorToArmSME/VectorToArmSMEPass.cpp
+++ b/mlir/lib/Conversion/VectorToArmSME/VectorToArmSMEPass.cpp
@@ -34,3 +34,7 @@ void ConvertVectorToArmSMEPass::runOnOperation() {
 
   (void)applyPatternsAndFoldGreedily(getOperation(), std::move(patterns));
 }
+
+std::unique_ptr<Pass> mlir::createConvertVectorToArmSMEPass() {
+  return std::make_unique<ConvertVectorToArmSMEPass>();
+}