================ @@ -1783,6 +1783,98 @@ void collectMapDataFromMapOperands(MapInfoData &mapData, } } +static int getMapDataMemberIdx(MapInfoData &mapData, + mlir::omp::MapInfoOp memberOp) { + int memberDataIdx = -1; + for (size_t i = 0; i < mapData.MapClause.size(); ++i) { + if (mapData.MapClause[i] == memberOp) + memberDataIdx = i; + } + return memberDataIdx; +} + +static mlir::omp::MapInfoOp +getFirstOrLastMappedMemberPtr(mlir::omp::MapInfoOp mapInfo, bool first) { + // Only 1 member has been mapped, we can return it. + if (mapInfo.getMembersIndex()->size() == 1) + if (auto mapOp = mlir::dyn_cast<mlir::omp::MapInfoOp>( + mapInfo.getMembers()[0].getDefiningOp())) + return mapOp; + + int64_t curPos = + mapInfo.getMembersIndex()->begin()->cast<mlir::IntegerAttr>().getInt(); + + int64_t idx = 1, curIdx = 0, memberPlacement = 0; + for (const auto *iter = std::next(mapInfo.getMembersIndex()->begin()); + iter != mapInfo.getMembersIndex()->end(); iter++) { + memberPlacement = iter->cast<mlir::IntegerAttr>().getInt(); + if (first) { + if (memberPlacement < curPos) { + curIdx = idx; + curPos = memberPlacement; + } + } else { + if (memberPlacement > curPos) { + curIdx = idx; + curPos = memberPlacement; + } + } + idx++; + } ---------------- agozillon wrote:
I imagine the former of sorting it when the function is called should be fine, I'll have to double check though. The latter, I don't think is possible, or at the very least I'd like to maintain the user-ordering for the maps for the moment (and member indices by extension) as I think it is important specification wise (at least until we can rule it out in any case) as there's only one case where the specification calls for re-ordering (which we still need to add support for actually) and that's re-ordering from/tofrom/to map clauses in-front of alloca/delete/release. https://github.com/llvm/llvm-project/pull/81510 _______________________________________________ llvm-branch-commits mailing list llvm-branch-commits@lists.llvm.org https://lists.llvm.org/cgi-bin/mailman/listinfo/llvm-branch-commits