[OpenMP] Fix target data use_device mapping bug (#111)

TIFitis · web-flow · commit d5a3c4dc3bb9 · 2024-07-08T19:01:50.000-05:00
diff --git a/mlir/lib/Target/LLVMIR/Dialect/OpenMP/OpenMPToLLVMIRTranslation.cpp b/mlir/lib/Target/LLVMIR/Dialect/OpenMP/OpenMPToLLVMIRTranslation.cpp
@@ -2230,7 +2230,7 @@ void collectMapDataFromMapOperands(
   auto findMapInfo = [&mapData](llvm::Value *val, unsigned &index) {
     index = 0;
     for (llvm::Value *basePtr : mapData.OriginalValue) {
-      if (basePtr == val)
+      if (basePtr == val && !mapData.IsAMember[index])
         return true;
       index++;
     }

Original file line number	Diff line number	Diff line change
`@@ -2230,7 +2230,7 @@ void collectMapDataFromMapOperands(`
`2230`	`2230`	`auto findMapInfo = [&mapData](llvm::Value *val, unsigned &index) {`
`2231`	`2231`	`index = 0;`
`2232`	`2232`	`for (llvm::Value *basePtr : mapData.OriginalValue) {`
`2233`		`- if (basePtr == val)`
	`2233`	`+ if (basePtr == val && !mapData.IsAMember[index])`
`2234`	`2234`	`return true;`
`2235`	`2235`	`index++;`
`2236`	`2236`	`}`