set exact flags on ConvertOpToUnsigned

amd-eochoalo · amd-eochoalo · commit 4271cb520815 · 2026-03-05T10:13:07.000-05:00
diff --git a/compiler/src/iree/compiler/Dialect/Util/Transforms/OptimizeIntArithmetic.cpp b/compiler/src/iree/compiler/Dialect/Util/Transforms/OptimizeIntArithmetic.cpp
@@ -102,8 +102,11 @@ struct ConvertOpToUnsigned : public OpRewritePattern<Signed> {
     if (failed(staticallyLegalToConvertToUnsignedOp(solver, op))) {
       return failure();
     }
-    rewriter.replaceOpWithNewOp<Unsigned>(op, op->getResultTypes(),
+    auto newOp = rewriter.replaceOpWithNewOp<Unsigned>(op, op->getResultTypes(),
                                           op->getOperands(), op->getAttrs());
+    if constexpr (std::is_same_v<Unsigned, arith::IndexCastUIOp>) {
+      newOp.setExact(true);
+    }
     return success();
   }
 

Original file line number	Diff line number	Diff line change
`@@ -102,8 +102,11 @@ struct ConvertOpToUnsigned : public OpRewritePattern<Signed> {`
`102`	`102`	`if (failed(staticallyLegalToConvertToUnsignedOp(solver, op))) {`
`103`	`103`	`return failure();`
`104`	`104`	`}`
`105`		`- rewriter.replaceOpWithNewOp<Unsigned>(op, op->getResultTypes(),`
	`105`	`+ auto newOp = rewriter.replaceOpWithNewOp<Unsigned>(op, op->getResultTypes(),`
`106`	`106`	`op->getOperands(), op->getAttrs());`
	`107`	`+ if constexpr (std::is_same_v<Unsigned, arith::IndexCastUIOp>) {`
	`108`	`+ newOp.setExact(true);`
	`109`	`+ }`
`107`	`110`	`return success();`
`108`	`111`	`}`
`109`	`112`