aboutsummaryrefslogtreecommitdiff
path: root/mlir/lib/Dialect/Utils/StaticValueUtils.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'mlir/lib/Dialect/Utils/StaticValueUtils.cpp')
-rw-r--r--mlir/lib/Dialect/Utils/StaticValueUtils.cpp6
1 files changed, 5 insertions, 1 deletions
diff --git a/mlir/lib/Dialect/Utils/StaticValueUtils.cpp b/mlir/lib/Dialect/Utils/StaticValueUtils.cpp
index 1cded38..e6ef028 100644
--- a/mlir/lib/Dialect/Utils/StaticValueUtils.cpp
+++ b/mlir/lib/Dialect/Utils/StaticValueUtils.cpp
@@ -181,12 +181,16 @@ bool isEqualConstantIntOrValueArray(ArrayRef<OpFoldResult> ofrs1,
return true;
}
-/// Return a vector of OpFoldResults with the same size a staticValues, but all
+/// Return a vector of OpFoldResults with the same size as staticValues, but all
/// elements for which ShapedType::isDynamic is true, will be replaced by
/// dynamicValues.
SmallVector<OpFoldResult> getMixedValues(ArrayRef<int64_t> staticValues,
ValueRange dynamicValues,
MLIRContext *context) {
+ assert(dynamicValues.size() == static_cast<size_t>(llvm::count_if(
+ staticValues, ShapedType::isDynamic)) &&
+ "expected the rank of dynamic values to match the number of "
+ "values known to be dynamic");
SmallVector<OpFoldResult> res;
res.reserve(staticValues.size());
unsigned numDynamic = 0;