Fix debug build?

qedawkins · qedawkins · commit 83476ee43e7d · 2023-10-22T21:03:07.000-04:00
diff --git a/compiler/src/iree/compiler/Utils/CustomPatternApplicatorPassBase.cpp b/compiler/src/iree/compiler/Utils/CustomPatternApplicatorPassBase.cpp
@@ -56,9 +56,9 @@ static SmallVector<Value> getI32TensorSizes(PatternRewriter &rewriter,
 }
 
 static FailureOr<Value> extractValueFromRange(PatternRewriter &rewriter,
-                                              ValueRange vals,
-                                              IntegerAttr index) {
-  if (index.getInt() >= vals.size())
+                                              ValueRange vals, Attribute attr) {
+  IntegerAttr index = dyn_cast<IntegerAttr>(attr);
+  if (!index || index.getInt() >= vals.size())
     return failure();
   return vals[index.getInt()];
 }

Original file line number	Diff line number	Diff line change
`@@ -56,9 +56,9 @@ static SmallVector<Value> getI32TensorSizes(PatternRewriter &rewriter,`
`56`	`56`	`}`
`57`	`57`
`58`	`58`	`static FailureOr<Value> extractValueFromRange(PatternRewriter &rewriter,`
`59`		`- ValueRange vals,`
`60`		`- IntegerAttr index) {`
`61`		`- if (index.getInt() >= vals.size())`
	`59`	`+ ValueRange vals, Attribute attr) {`
	`60`	`+ IntegerAttr index = dyn_cast<IntegerAttr>(attr);`
	`61`	`+ if (!index \|\| index.getInt() >= vals.size())`
`62`	`62`	`return failure();`
`63`	`63`	`return vals[index.getInt()];`
`64`	`64`	`}`