================
@@ -58,21 +67,62 @@ class MapInfoFinalizationPass
/*corresponding local alloca=*/fir::AllocaOp>
localBoxAllocas;
- void genDescriptorMemberMaps(mlir::omp::MapInfoOp op,
- fir::FirOpBuilder &builder,
- mlir::Operation *target) {
- mlir::Location loc = op.getLoc();
- mlir::Value descriptor = op.getVarPtr();
+ /// getMemberUserList gathers all users of a particular MapInfoOp that are
+ /// other MapInfoOp's and places them into the mapMemberUsers list, which
+ /// records the map that the current argument MapInfoOp "op" is part of
+ /// alongside the placement of "op" in the recorded users members list. The
+ /// intent of the generated list is to find all MapInfoOp's that may be
+ /// considered parents of the passed in "op" and in which it shows up in the
+ /// member list, alongside collecting the placement information of "op" in
its
+ /// parents member list.
+ void
+ getMemberUserList(mlir::omp::MapInfoOp op,
+ llvm::SmallVectorImpl<ParentAndPlacement> &mapMemberUsers)
{
+ for (auto *users : op->getUsers())
+ if (auto map = mlir::dyn_cast_if_present<mlir::omp::MapInfoOp>(users))
+ for (auto [i, mapMember] : llvm::enumerate(map.getMembers()))
+ if (mapMember.getDefiningOp() == op)
+ mapMemberUsers.push_back({map, i});
+ }
+
+ llvm::SmallVector<int64_t>
+ getAsIntegers(llvm::ArrayRef<mlir::Attribute> values) {
+ llvm::SmallVector<int64_t> ints;
+ ints.reserve(values.size());
+ llvm::transform(values, std::back_inserter(ints),
+ [](mlir::Attribute value) {
+ return mlir::cast<mlir::IntegerAttr>(value).getInt();
+ });
+ return ints;
+ }
+
+ /// This function will expand a MapInfoOp's member indices back into a vector
+ /// so that they can be trivially modified as unfortunately the attribute
type
+ /// that's used does not have modifiable fields at the moment (generally
+ /// awkward to work with)
+ void getMemberIndicesAsVectors(
+ mlir::omp::MapInfoOp mapInfo,
+ llvm::SmallVector<llvm::SmallVector<int64_t>> &indices) {
----------------
skatrak wrote:
```suggestion
llvm::SmallVectorImpl<llvm::SmallVector<int64_t>> &indices) {
```
https://github.com/llvm/llvm-project/pull/111192
_______________________________________________
llvm-branch-commits mailing list
llvm-branch-commits@lists.llvm.org
https://lists.llvm.org/cgi-bin/mailman/listinfo/llvm-branch-commits
