diff --git a/mlir/lib/Dialect/Linalg/Transforms/Specialize.cpp b/mlir/lib/Dialect/Linalg/Transforms/Specialize.cpp index 35ba4f159113f..40fc0d68e358f 100644 --- a/mlir/lib/Dialect/Linalg/Transforms/Specialize.cpp +++ b/mlir/lib/Dialect/Linalg/Transforms/Specialize.cpp @@ -182,11 +182,9 @@ static FailureOr specializeLinalgContractions(RewriterBase &rewriter, if (!mlir::linalg::detail::isContractionBody( *genericOp.getBlock(), [](Operation *first, Operation *second) { - if ((isa(first) && isa(second)) || - (isa(first) && isa(second)) || - (isa(first) && isa(second))) - return true; - return false; + return (isa(first) && isa(second)) || + (isa(first) && isa(second)) || + (isa(first) && isa(second)); })) return failure();