blob: c70b5f000107a590db021f3e9bb9711a16b1ed5a (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
//===- ToLLVMInterface.cpp - MLIR LLVM Conversion -------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#include "mlir/Conversion/ConvertToLLVM/ToLLVMInterface.h"
#include "mlir/IR/Dialect.h"
#include "mlir/IR/Operation.h"
using namespace mlir;
void mlir::populateConversionTargetFromOperation(
Operation *root, ConversionTarget &target, LLVMTypeConverter &typeConverter,
RewritePatternSet &patterns) {
DenseSet<Dialect *> dialects;
root->walk([&](Operation *op) {
Dialect *dialect = op->getDialect();
if (!dialects.insert(dialect).second)
return;
// First time we encounter this dialect: if it implements the interface,
// let's populate patterns !
auto *iface = dyn_cast<ConvertToLLVMPatternInterface>(dialect);
if (!iface)
return;
iface->populateConvertToLLVMConversionPatterns(target, typeConverter,
patterns);
});
}
void mlir::populateOpConvertToLLVMConversionPatterns(
Operation *op, ConversionTarget &target, LLVMTypeConverter &typeConverter,
RewritePatternSet &patterns) {
auto iface = dyn_cast<ConvertToLLVMOpInterface>(op);
if (!iface)
iface = op->getParentOfType<ConvertToLLVMOpInterface>();
if (!iface)
return;
SmallVector<ConvertToLLVMAttrInterface, 12> attrs;
iface.getConvertToLLVMConversionAttrs(attrs);
for (ConvertToLLVMAttrInterface attr : attrs)
attr.populateConvertToLLVMConversionPatterns(target, typeConverter,
patterns);
}
#include "mlir/Conversion/ConvertToLLVM/ToLLVMAttrInterface.cpp.inc"
#include "mlir/Conversion/ConvertToLLVM/ToLLVMOpInterface.cpp.inc"
|
