51 lines · cpp
1//===- ToLLVMInterface.cpp - MLIR LLVM Conversion -------------------------===//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-exception6//7//===----------------------------------------------------------------------===//8 9#include "mlir/Conversion/ConvertToLLVM/ToLLVMInterface.h"10#include "mlir/IR/Dialect.h"11#include "mlir/IR/Operation.h"12 13using namespace mlir;14 15void mlir::populateConversionTargetFromOperation(16 Operation *root, ConversionTarget &target, LLVMTypeConverter &typeConverter,17 RewritePatternSet &patterns) {18 DenseSet<Dialect *> dialects;19 root->walk([&](Operation *op) {20 Dialect *dialect = op->getDialect();21 if (!dialects.insert(dialect).second)22 return;23 // First time we encounter this dialect: if it implements the interface,24 // let's populate patterns !25 auto *iface = dyn_cast<ConvertToLLVMPatternInterface>(dialect);26 if (!iface)27 return;28 iface->populateConvertToLLVMConversionPatterns(target, typeConverter,29 patterns);30 });31}32 33void mlir::populateOpConvertToLLVMConversionPatterns(34 Operation *op, ConversionTarget &target, LLVMTypeConverter &typeConverter,35 RewritePatternSet &patterns) {36 auto iface = dyn_cast<ConvertToLLVMOpInterface>(op);37 if (!iface)38 iface = op->getParentOfType<ConvertToLLVMOpInterface>();39 if (!iface)40 return;41 SmallVector<ConvertToLLVMAttrInterface, 12> attrs;42 iface.getConvertToLLVMConversionAttrs(attrs);43 for (ConvertToLLVMAttrInterface attr : attrs)44 attr.populateConvertToLLVMConversionPatterns(target, typeConverter,45 patterns);46}47 48#include "mlir/Conversion/ConvertToLLVM/ToLLVMAttrInterface.cpp.inc"49 50#include "mlir/Conversion/ConvertToLLVM/ToLLVMOpInterface.cpp.inc"51