[flang-commits] [flang] [flang][cuda] Flag globals used in device function (PR #109460)
Slava Zakharin via flang-commits
flang-commits at lists.llvm.org
Fri Sep 20 12:14:54 PDT 2024
================
@@ -0,0 +1,73 @@
+//===-- CufOpConversion.cpp -----------------------------------------------===//
+//
+// 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 "flang/Common/Fortran.h"
+#include "flang/Optimizer/Dialect/CUF/CUFOps.h"
+#include "flang/Optimizer/Dialect/FIRDialect.h"
+#include "flang/Optimizer/Dialect/FIROps.h"
+#include "flang/Optimizer/HLFIR/HLFIROps.h"
+#include "flang/Runtime/CUDA/common.h"
+#include "flang/Runtime/allocatable.h"
+#include "mlir/Pass/Pass.h"
+#include "mlir/Transforms/DialectConversion.h"
+
+namespace fir {
+#define GEN_PASS_DEF_CUFIMPLICITDEVICEGLOBAL
+#include "flang/Optimizer/Transforms/Passes.h.inc"
+} // namespace fir
+
+namespace {
+
+static fir::GlobalOp getGlobalOpFromValue(mlir::Value v) {
+ if (auto addrOfOp{mlir::dyn_cast_or_null<fir::AddrOfOp>(v.getDefiningOp())}) {
+ auto sym{mlir::SymbolTable::lookupNearestSymbolFrom(
+ addrOfOp, addrOfOp.getSymbolAttr())};
+ return mlir::dyn_cast_or_null<fir::GlobalOp>(sym);
+ }
+ return nullptr;
+}
+
+static void prepareImplicitDeviceGlobals(mlir::func::FuncOp funcOp,
+ bool onlyConstant = true) {
+ auto cudaProcAttr{
+ funcOp->getAttrOfType<cuf::ProcAttributeAttr>(cuf::getProcAttrName())};
+ if (!cudaProcAttr || cudaProcAttr.getValue() == cuf::ProcAttribute::Host)
+ return;
+ for (auto addrOfOp : funcOp.getBody().getOps<fir::AddrOfOp>()) {
+ if (auto globalOp{getGlobalOpFromValue(addrOfOp.getResult())}) {
+ bool isCandidate{(onlyConstant ? globalOp.getConstant() : true) &&
+ !globalOp.getDataAttr()};
+ if (isCandidate)
+ globalOp.setDataAttrAttr(cuf::DataAttributeAttr::get(
+ funcOp.getContext(), globalOp.getConstant()
+ ? cuf::DataAttribute::Constant
+ : cuf::DataAttribute::Device));
+ }
+ }
+}
+
+class CufImplicitDeviceGlobal
+ : public fir::impl::CufImplicitDeviceGlobalBase<CufImplicitDeviceGlobal> {
+public:
+ void runOnOperation() override {
+ auto *ctx = &getContext();
+ mlir::RewritePatternSet patterns(ctx);
+ mlir::ConversionTarget target(*ctx);
----------------
vzakhari wrote:
```suggestion
```
https://github.com/llvm/llvm-project/pull/109460
More information about the flang-commits
mailing list