[OpenMP] Fix target data use_device mapping bug

ROCm · Jul 8, 2024 · 970e711 · 970e711
1 parent 0ef3716
commit 970e711
Showing 1 changed file with 1 addition and 1 deletion.
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++;
     }