Fix duplicate mapping detection in gpu::setMappingAttr() (#77499)

RoboTux · web-flow · commit 76e79b0bef6c · 2024-02-20T09:54:00.000Z
diff --git a/mlir/lib/Dialect/GPU/Transforms/ParallelLoopMapper.cpp b/mlir/lib/Dialect/GPU/Transforms/ParallelLoopMapper.cpp
@@ -41,6 +41,7 @@ gpu::setMappingAttr(ParallelOp ploopOp,
         specifiedMappings.count(processor))
       return ploopOp.emitError(
           "invalid mapping multiple loops to same processor");
+    specifiedMappings.insert(processor);
   }
   ArrayRef<Attribute> mappingAsAttrs(mapping.data(), mapping.size());
   ploopOp->setAttr(getMappingAttrName(),

Original file line number	Diff line number	Diff line change
`@@ -41,6 +41,7 @@ gpu::setMappingAttr(ParallelOp ploopOp,`
`41`	`41`	`specifiedMappings.count(processor))`
`42`	`42`	`return ploopOp.emitError(`
`43`	`43`	`"invalid mapping multiple loops to same processor");`
	`44`	`+ specifiedMappings.insert(processor);`
`44`	`45`	`}`
`45`	`46`	`ArrayRef<Attribute> mappingAsAttrs(mapping.data(), mapping.size());`
`46`	`47`	`ploopOp->setAttr(getMappingAttrName(),`