MLIR  22.0.0git
ArmSVEVectorTransformOps.cpp
Go to the documentation of this file.
1 //===- ArmSVEVectorTransformOps.cpp - Implementation transform ops -------===//
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 
10 
14 
15 using namespace mlir;
16 
17 //===----------------------------------------------------------------------===//
18 // Apply...PatternsOp
19 //===----------------------------------------------------------------------===//
20 
21 void transform::ApplyArmSVELowerContractionToI8MMPatternsOp::populatePatterns(
24 }
25 
26 void transform::ApplyArmSVELowerContractionToBFMMLAPatternsOp::populatePatterns(
29 }
30 
31 //===----------------------------------------------------------------------===//
32 // Transform op registration
33 //===----------------------------------------------------------------------===//
34 
35 namespace {
36 class ArmSVEVectorTransformDialectExtension
38  ArmSVEVectorTransformDialectExtension> {
39 public:
41  ArmSVEVectorTransformDialectExtension)
42 
43  ArmSVEVectorTransformDialectExtension() {
44  declareGeneratedDialect<arm_sve::ArmSVEDialect>();
45  registerTransformOps<
46 #define GET_OP_LIST
47 #include "mlir/Dialect/ArmSVE/TransformOps/ArmSVEVectorTransformOps.cpp.inc"
48  >();
49  }
50 };
51 } // namespace
52 
53 #define GET_OP_CLASSES
54 #include "mlir/Dialect/ArmSVE/TransformOps/ArmSVEVectorTransformOps.cpp.inc"
55 
57  DialectRegistry &registry) {
58  registry.addExtensions<ArmSVEVectorTransformDialectExtension>();
59 }
#define MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(CLASS_NAME)
Definition: TypeID.h:331
The DialectRegistry maps a dialect namespace to a constructor for the matching dialect.
void addExtensions()
Add the given extensions to the registry.
Base class for extensions of the Transform dialect that supports injecting operations into the Transf...
void registerTransformDialectExtension(DialectRegistry &registry)
Include the generated interface declarations.
const FrozenRewritePatternSet & patterns
void populateLowerContractionToSVEBFMMLAPatterns(RewritePatternSet &patterns)
void populateLowerContractionToSVEI8MMPatterns(RewritePatternSet &patterns)