aboutsummaryrefslogtreecommitdiff
path: root/mlir/lib/Dialect/Arith/Transforms/ReifyValueBounds.cpp
diff options
context:
space:
mode:
authorVitaly Buka <vitalybuka@google.com>2024-04-04 17:49:07 -0700
committerVitaly Buka <vitalybuka@google.com>2024-04-04 17:49:07 -0700
commita724510541fc3272c9d4415c89b4549d8d149675 (patch)
tree5090317c71cf2ae73fb91a32f8dd6f8e037e4603 /mlir/lib/Dialect/Arith/Transforms/ReifyValueBounds.cpp
parent2fe88fc8b7a3c27d473b6a172f0dc8aae7be3310 (diff)
parentb76eb1ddfbacda273b8e6a9940f1da6812fdc2e0 (diff)
downloadllvm-users/vitalybuka/spr/main.rename-remove-traps-to-lower-builtin-hot.zip
llvm-users/vitalybuka/spr/main.rename-remove-traps-to-lower-builtin-hot.tar.gz
llvm-users/vitalybuka/spr/main.rename-remove-traps-to-lower-builtin-hot.tar.bz2
[𝘀𝗽𝗿] changes introduced through rebaseusers/vitalybuka/spr/main.rename-remove-traps-to-lower-builtin-hot
Created using spr 1.3.4 [skip ci]
Diffstat (limited to 'mlir/lib/Dialect/Arith/Transforms/ReifyValueBounds.cpp')
-rw-r--r--mlir/lib/Dialect/Arith/Transforms/ReifyValueBounds.cpp6
1 files changed, 4 insertions, 2 deletions
diff --git a/mlir/lib/Dialect/Arith/Transforms/ReifyValueBounds.cpp b/mlir/lib/Dialect/Arith/Transforms/ReifyValueBounds.cpp
index 8d9fd14..fad2212 100644
--- a/mlir/lib/Dialect/Arith/Transforms/ReifyValueBounds.cpp
+++ b/mlir/lib/Dialect/Arith/Transforms/ReifyValueBounds.cpp
@@ -119,7 +119,8 @@ FailureOr<OpFoldResult> mlir::arith::reifyShapedValueDimBound(
OpBuilder &b, Location loc, presburger::BoundType type, Value value,
int64_t dim, ValueBoundsConstraintSet::StopConditionFn stopCondition,
bool closedUB) {
- auto reifyToOperands = [&](Value v, std::optional<int64_t> d) {
+ auto reifyToOperands = [&](Value v, std::optional<int64_t> d,
+ ValueBoundsConstraintSet &cstr) {
// We are trying to reify a bound for `value` in terms of the owning op's
// operands. Construct a stop condition that evaluates to "true" for any SSA
// value expect for `value`. I.e., the bound will be computed in terms of
@@ -135,7 +136,8 @@ FailureOr<OpFoldResult> mlir::arith::reifyShapedValueDimBound(
FailureOr<OpFoldResult> mlir::arith::reifyIndexValueBound(
OpBuilder &b, Location loc, presburger::BoundType type, Value value,
ValueBoundsConstraintSet::StopConditionFn stopCondition, bool closedUB) {
- auto reifyToOperands = [&](Value v, std::optional<int64_t> d) {
+ auto reifyToOperands = [&](Value v, std::optional<int64_t> d,
+ ValueBoundsConstraintSet &cstr) {
return v != value;
};
return reifyValueBound(b, loc, type, value, /*dim=*/std::nullopt,