[AMDGPU] Fix missing IsExact flag when expanding vector binary operator

shiltian · shiltian · commit 57eba6f8eb92 · 2024-03-26T15:27:06.000-04:00
diff --git a/llvm/lib/Target/AMDGPU/AMDGPUCodeGenPrepare.cpp b/llvm/lib/Target/AMDGPU/AMDGPUCodeGenPrepare.cpp
@@ -1594,6 +1594,11 @@ bool AMDGPUCodeGenPrepareImpl::visitBinaryOperator(BinaryOperator &I) {
           }
         }
 
+        if (auto *NewEltI = dyn_cast<Instruction>(NewElt))
+          if (PossiblyExactOperator::isPossiblyExactOpcode(
+                  NewEltI->getOpcode()))
+            NewEltI->setIsExact(I.isExact());
+
         NewDiv = Builder.CreateInsertElement(NewDiv, NewElt, N);
       }
     } else {

Original file line number	Diff line number	Diff line change
`@@ -1594,6 +1594,11 @@ bool AMDGPUCodeGenPrepareImpl::visitBinaryOperator(BinaryOperator &I) {`
`1594`	`1594`	`}`
`1595`	`1595`	`}`
`1596`	`1596`
	`1597`	`+ if (auto *NewEltI = dyn_cast<Instruction>(NewElt))`
	`1598`	`+ if (PossiblyExactOperator::isPossiblyExactOpcode(`
	`1599`	`+ NewEltI->getOpcode()))`
	`1600`	`+ NewEltI->setIsExact(I.isExact());`
	`1601`	`+`
`1597`	`1602`	`NewDiv = Builder.CreateInsertElement(NewDiv, NewElt, N);`
`1598`	`1603`	`}`
`1599`	`1604`	`} else {`