1 //===- VectorToArmSME.h - Convert vector to ArmSME dialect ----------------===// 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 #ifndef MLIR_CONVERSION_VECTORTOARMSME_VECTORTOARMSME_H_ 9 #define MLIR_CONVERSION_VECTORTOARMSME_VECTORTOARMSME_H_ 10 11 #include "mlir/IR/PatternMatch.h" 12 13 namespace mlir { 14 class Pass; 15 16 #define GEN_PASS_DECL_CONVERTVECTORTOARMSME 17 #include "mlir/Conversion/Passes.h.inc" 18 19 /// Collect a set of patterns to lower Vector ops to ArmSME ops that map to LLVM 20 /// intrinsics. 21 void populateVectorToArmSMEPatterns(RewritePatternSet &patterns, 22 MLIRContext &ctx); 23 24 /// Create a pass to lower operations from the vector dialect to Arm SME. 25 std::unique_ptr<Pass> createConvertVectorToArmSMEPass(); 26 27 } // namespace mlir 28 29 #endif // MLIR_CONVERSION_VECTORTOARMSME_VECTORTOARMSME_H_ 30