MLIR  19.0.0git
Transforms.h
Go to the documentation of this file.
1 //===- Transforms.h - AMX Dialect Transformation Entrypoints ----*- C++ -*-===//
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 
9 #ifndef MLIR_DIALECT_AMX_TRANSFORMS_H
10 #define MLIR_DIALECT_AMX_TRANSFORMS_H
11 
12 namespace mlir {
13 
14 class LLVMConversionTarget;
15 class LLVMTypeConverter;
16 class RewritePatternSet;
17 
18 /// Collect a set of patterns to lower AMX ops to ops that map to LLVM
19 /// intrinsics.
20 void populateAMXLegalizeForLLVMExportPatterns(LLVMTypeConverter &converter,
21  RewritePatternSet &patterns);
22 
23 /// Configure the target to support lowering AMX ops to ops that map to LLVM
24 /// intrinsics.
25 void configureAMXLegalizeForExportTarget(LLVMConversionTarget &target);
26 
27 } // namespace mlir
28 
29 #endif // MLIR_DIALECT_AMX_TRANSFORMS_H
Include the generated interface declarations.
void populateAMXLegalizeForLLVMExportPatterns(LLVMTypeConverter &converter, RewritePatternSet &patterns)
Collect a set of patterns to lower AMX ops to ops that map to LLVM intrinsics.
void configureAMXLegalizeForExportTarget(LLVMConversionTarget &target)
Configure the target to support lowering AMX ops to ops that map to LLVM intrinsics.