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

shiltian · shiltian · commit 6a4406f64994 · 2024-03-26T14:47:19.000-04:00
diff --git a/llvm/lib/Target/AMDGPU/AMDGPUCodeGenPrepare.cpp b/llvm/lib/Target/AMDGPU/AMDGPUCodeGenPrepare.cpp
@@ -1594,6 +1594,9 @@ bool AMDGPUCodeGenPrepareImpl::visitBinaryOperator(BinaryOperator &I) {
           }
         }
 
+        if (auto *NewEltI = dyn_cast<Instruction>(NewElt))
+          NewEltI->setIsExact(I.isExact());
+
         NewDiv = Builder.CreateInsertElement(NewDiv, NewElt, N);
       }
     } else {

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