"unexpected opKind");
switch (expr.getKind()) {
case AffineExprKind::Constant:
- if (expr.cast<AffineConstantExpr>().getValue())
- return false;
- return true;
+ return expr.cast<AffineConstantExpr>().getValue() == 0;
case AffineExprKind::DimId:
return false;
case AffineExprKind::SymbolId:
auto resultType = op.result().getType().cast<ShapedType>();
constexpr int64_t kConstantFoldingMaxNumElements = 1024;
- if (resultType.getNumElements() > kConstantFoldingMaxNumElements)
- return false;
-
- return true;
+ return resultType.getNumElements() <= kConstantFoldingMaxNumElements;
};
tensor::populateFoldConstantExtractSlicePatterns(patterns, controlFn);