compact resize op lowering

2026-05-15 17:36:12 +02:00
parent fe244d5aa1
commit 78242e2887
3 changed files with 95 additions and 38 deletions
@@ -110,6 +110,14 @@ llvm::FailureOr<int64_t> resolveIndexValueImpl(mlir::Value value, const StaticVa
    return static_cast<int64_t>(static_cast<uint64_t>(*lhs) / static_cast<uint64_t>(*rhs));
  }

+  if (auto minOp = mlir::dyn_cast<mlir::arith::MinUIOp>(definingOp)) {
+    auto lhs = resolveIndexValueImpl(minOp.getLhs(), knowledge);
+    auto rhs = resolveIndexValueImpl(minOp.getRhs(), knowledge);
+    if (failed(lhs) || failed(rhs))
+      return mlir::failure();
+    return static_cast<int64_t>(std::min(static_cast<uint64_t>(*lhs), static_cast<uint64_t>(*rhs)));
+  }
+
  if (auto remOp = mlir::dyn_cast<mlir::arith::RemUIOp>(definingOp)) {
    auto lhs = resolveIndexValueImpl(remOp.getLhs(), knowledge);
    auto rhs = resolveIndexValueImpl(remOp.getRhs(), knowledge);
@@ -12,6 +12,7 @@ bool isCoreStaticAddressOp(mlir::Operation* op) {
                   mlir::arith::SubIOp,
                   mlir::arith::MulIOp,
                   mlir::arith::DivUIOp,
+                   mlir::arith::MinUIOp,
                   mlir::arith::RemUIOp,
                   mlir::arith::IndexCastOp,
                   mlir::memref::AllocOp,