[flang-commits] [flang] [flang][cuda] Expose conversion patterns from CUF to FIR calls (PR #109465)
Valentin Clement バレンタイン クレメン via flang-commits
flang-commits at lists.llvm.org
Fri Sep 20 13:06:27 PDT 2024
https://github.com/clementval created https://github.com/llvm/llvm-project/pull/109465
Expose conversion patterns so they can be reused outside of this pass.
>From 7f10e7ac0e341db522cd6f99f3ad2cc92e62c64b Mon Sep 17 00:00:00 2001
From: Valentin Clement <clementval at gmail.com>
Date: Fri, 20 Sep 2024 13:05:16 -0700
Subject: [PATCH] [flang][cuda] Expose conversion patterns from CUF to FIR
calls
Expose conversion patterns so they can be reused outside of this pass.
---
flang/lib/Optimizer/Transforms/CufOpConversion.cpp | 14 +++++++++++---
1 file changed, 11 insertions(+), 3 deletions(-)
diff --git a/flang/lib/Optimizer/Transforms/CufOpConversion.cpp b/flang/lib/Optimizer/Transforms/CufOpConversion.cpp
index ac796e83b07078..f8ace2dd96a0d8 100644
--- a/flang/lib/Optimizer/Transforms/CufOpConversion.cpp
+++ b/flang/lib/Optimizer/Transforms/CufOpConversion.cpp
@@ -6,6 +6,7 @@
//
//===----------------------------------------------------------------------===//
+#include "flang/Optimizer/Transforms/CufOpConversion.h"
#include "flang/Common/Fortran.h"
#include "flang/Optimizer/Builder/Runtime/RTBuilder.h"
#include "flang/Optimizer/CodeGen/TypeConverter.h"
@@ -473,9 +474,7 @@ class CufOpConversion : public fir::impl::CufOpConversionBase<CufOpConversion> {
!mlir::isa<fir::BaseBoxType>(dstTy);
});
target.addLegalDialect<fir::FIROpsDialect, mlir::arith::ArithDialect>();
- patterns.insert<CufAllocOpConversion>(ctx, &*dl, &typeConverter);
- patterns.insert<CufAllocateOpConversion, CufDeallocateOpConversion,
- CufFreeOpConversion, CufDataTransferOpConversion>(ctx);
+ cuf::populateCUFToFIRConversionPatterns(typeConverter, *dl, patterns);
if (mlir::failed(mlir::applyPartialConversion(getOperation(), target,
std::move(patterns)))) {
mlir::emitError(mlir::UnknownLoc::get(ctx),
@@ -485,3 +484,12 @@ class CufOpConversion : public fir::impl::CufOpConversionBase<CufOpConversion> {
}
};
} // namespace
+
+void cuf::populateCUFToFIRConversionPatterns(
+ fir::LLVMTypeConverter &converter, mlir::DataLayout &dl,
+ mlir::RewritePatternSet &patterns) {
+ patterns.insert<CufAllocOpConversion>(patterns.getContext(), &dl, &converter);
+ patterns.insert<CufAllocateOpConversion, CufDeallocateOpConversion,
+ CufFreeOpConversion, CufDataTransferOpConversion>(
+ patterns.getContext());
+}
More information about the flang-commits
mailing list