[Mlir-commits] [mlir] merge code for llvm.emit_c_interface into convertFuncOpToLLVMFuncOp (PR #92986)
llvmlistbot at llvm.org
llvmlistbot at llvm.org
Thu May 23 09:45:57 PDT 2024
================
@@ -0,0 +1,83 @@
+//===- TestConvertFuncOp.cpp - Test LLVM Conversion of Func FuncOp --------===//
+//
+// 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 "TestDialect.h"
+
+#include "mlir/Conversion/FuncToLLVM/ConvertFuncToLLVM.h"
+#include "mlir/Conversion/LLVMCommon/ConversionTarget.h"
+#include "mlir/Conversion/LLVMCommon/Pattern.h"
+#include "mlir/Dialect/Func/IR/FuncOps.h"
+#include "mlir/Dialect/LLVMIR/LLVMDialect.h"
+#include "mlir/IR/PatternMatch.h"
+#include "mlir/Pass/Pass.h"
+
+using namespace mlir;
+
+namespace {
+
+/// Test helper Conversion Pattern to directly call `convertFuncOpToLLVMFuncOp`
+/// to verify this utility function includes all functionalities of conversion
+struct FuncOpConversion : public ConvertOpToLLVMPattern<func::FuncOp> {
+ FuncOpConversion(const LLVMTypeConverter &converter)
+ : ConvertOpToLLVMPattern(converter) {}
+
+ LogicalResult
+ matchAndRewrite(func::FuncOp funcOp, OpAdaptor adaptor,
+ ConversionPatternRewriter &rewriter) const override {
+ FailureOr<LLVM::LLVMFuncOp> newFuncOp = mlir::convertFuncOpToLLVMFuncOp(
+ cast<FunctionOpInterface>(funcOp.getOperation()), rewriter,
+ *getTypeConverter());
+ if (failed(newFuncOp))
+ return rewriter.notifyMatchFailure(funcOp, "Could not convert funcop");
+
+ rewriter.eraseOp(funcOp);
+ return success();
+ }
+};
+
+struct TestConvertFuncOp
+ : public PassWrapper<TestConvertFuncOp, OperationPass<ModuleOp>> {
+ MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(TestConvertFuncOp)
+
+ void getDependentDialects(DialectRegistry ®istry) const final {
+ registry.insert<LLVM::LLVMDialect>();
+ }
+
+ [[nodiscard]] StringRef getArgument() const final {
+ return "test-convert-func-op";
+ }
+
+ [[nodiscard]] StringRef getDescription() const final {
+ return "Tests conversion of `func.func` to `llvm.func` for different "
+ "attributes";
+ }
+
+ void runOnOperation() override {
+ MLIRContext *ctx = &getContext();
+
+ LowerToLLVMOptions options(ctx);
+ // Populate type conversions.
+ LLVMTypeConverter typeConverter(ctx, options);
+
+ RewritePatternSet patterns(ctx);
+ patterns.add<FuncOpConversion>(ctx);
+
+ populateFuncToLLVMConversionPatterns(typeConverter, patterns, nullptr);
----------------
fengxie wrote:
@ftynse if got a chance, can you review latest changes are good to address feedbacks?
* deleting general pattern and replace it with return op -> llvm.return pattern.
```c++
populateFuncToLLVMConversionPatterns(typeConverter, patterns, nullptr);
```
https://github.com/llvm/llvm-project/pull/92986
More information about the Mlir-commits
mailing list