[flang-commits] [flang] [OpenMP] Add TargetAMDGPU support for Complex argument and return types (PR #144924)
Akash Banerjee via flang-commits
flang-commits at lists.llvm.org
Thu Jun 19 09:31:36 PDT 2025
https://github.com/TIFitis created https://github.com/llvm/llvm-project/pull/144924
This patch adds TargetAMDGPU offloading support for Complex argument and return types.
>From 647c2ff815d4a07c86ec9667fea24a4f6490a707 Mon Sep 17 00:00:00 2001
From: Akash Banerjee <Akash.Banerjee at amd.com>
Date: Thu, 19 Jun 2025 17:23:45 +0100
Subject: [PATCH] [OpenMP] Add TargetAMDGPU support for Complex argument and
return types
This patch adds TargetAMDGPU offloading support for Complex argument and return types.
---
flang/lib/Optimizer/CodeGen/Target.cpp | 25 +++++++++++++++++++++++--
1 file changed, 23 insertions(+), 2 deletions(-)
diff --git a/flang/lib/Optimizer/CodeGen/Target.cpp b/flang/lib/Optimizer/CodeGen/Target.cpp
index 7dbf21ce0c125..b60a72e4340b9 100644
--- a/flang/lib/Optimizer/CodeGen/Target.cpp
+++ b/flang/lib/Optimizer/CodeGen/Target.cpp
@@ -1443,14 +1443,35 @@ struct TargetAMDGPU : public GenericTarget<TargetAMDGPU> {
CodeGenSpecifics::Marshalling
complexArgumentType(mlir::Location loc, mlir::Type eleTy) const override {
CodeGenSpecifics::Marshalling marshal;
- TODO(loc, "handle complex argument types");
+ const auto *sem = &floatToSemantics(kindMap, eleTy);
+ if (sem == &llvm::APFloat::IEEEsingle()) {
+ // Lower COMPLEX(KIND=4) as an array of two element values.
+ marshal.emplace_back(fir::SequenceType::get({2}, eleTy), AT{});
+ } else if (sem == &llvm::APFloat::IEEEdouble()) {
+ // Pass COMPLEX(KIND=8) as two separate arguments.
+ marshal.emplace_back(eleTy, AT{});
+ marshal.emplace_back(eleTy, AT{});
+ } else {
+ typeTodo(sem, loc, "argument");
+ }
return marshal;
}
CodeGenSpecifics::Marshalling
complexReturnType(mlir::Location loc, mlir::Type eleTy) const override {
CodeGenSpecifics::Marshalling marshal;
- TODO(loc, "handle complex return types");
+ const auto *sem = &floatToSemantics(kindMap, eleTy);
+ if (sem == &llvm::APFloat::IEEEsingle()) {
+ // Return COMPLEX(KIND=4) as an array of two elements.
+ marshal.emplace_back(fir::SequenceType::get({2}, eleTy), AT{});
+ } else if (sem == &llvm::APFloat::IEEEdouble()) {
+ // Return COMPLEX(KIND=8) via an aggregate with two fields.
+ marshal.emplace_back(mlir::TupleType::get(eleTy.getContext(),
+ mlir::TypeRange{eleTy, eleTy}),
+ AT{});
+ } else {
+ typeTodo(sem, loc, "return");
+ }
return marshal;
}
};
More information about the flang-commits
mailing list