shylie/llvm-project
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[mlir] Strip away lambdas (NFC) (#143280)

Browse Source 
We don't need lambdas here.
This commit is contained in:
Kazu Hirata 2025-06-08 01:34:17 -07:00 committed by GitHub
parent 9ea3972cd1
commit 1cf1c21b84
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 10 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mlir/lib/Dialect/Affine/Analysis/LoopAnalysis.cpp
View File

@ -402,9 +402,7 @@ isVectorizableLoopBodyWithOpCond(AffineForOp loop,
return !VectorType::isValidElementType(type);
}))
return true;
return llvm::any_of(op.getResultTypes(), [](Type type) {
return !VectorType::isValidElementType(type);
});
return !llvm::all_of(op.getResultTypes(), VectorType::isValidElementType);
});
SmallVector<NestedMatch, 8> opsMatched;
types.match(forOp, &opsMatched);

15
mlir/lib/Dialect/Linalg/Transforms/Vectorization.cpp
View File

@ -820,9 +820,8 @@ tensorExtractVectorizationPrecondition(Operation *op, bool vectorizeNDExtract) {
return failure();
}
if (llvm::any_of(extractOp->getResultTypes(), [](Type type) {
return !VectorType::isValidElementType(type);
})) {
if (!llvm::all_of(extractOp->getResultTypes(),
VectorType::isValidElementType)) {
return failure();
}
@ -2287,14 +2286,12 @@ static LogicalResult vectorizeLinalgOpPrecondition(
})) {
continue;
}
if (llvm::any_of(innerOp.getOperandTypes(), [](Type type) {
return !VectorType::isValidElementType(type);
})) {
if (!llvm::all_of(innerOp.getOperandTypes(),
VectorType::isValidElementType)) {
return failure();
}
if (llvm::any_of(innerOp.getResultTypes(), [](Type type) {
return !VectorType::isValidElementType(type);
})) {
if (!llvm::all_of(innerOp.getResultTypes(),
VectorType::isValidElementType)) {
return failure();
}
}

6
mlir/lib/Dialect/SparseTensor/IR/SparseTensorDialect.cpp
View File

@ -799,8 +799,7 @@ LogicalResult SparseTensorEncodingAttr::verify(
"before singleton level";
auto *curCOOEnd = std::find_if_not(it, lvlTypes.end(), isSingletonLT);
if (!std::all_of(it, curCOOEnd,
[](LevelType i) { return isSingletonLT(i); }))
if (!std::all_of(it, curCOOEnd, isSingletonLT))
return emitError() << "expected all singleton lvlTypes "
"following a singleton level";
// We can potentially support mixed SoA/AoS singleton levels.
@ -833,8 +832,7 @@ LogicalResult SparseTensorEncodingAttr::verify(
it != std::end(lvlTypes)) {
if (it != lvlTypes.end() - 1)
return emitError() << "expected n_out_of_m to be the last level type";
if (!std::all_of(lvlTypes.begin(), it,
[](LevelType i) { return isDenseLT(i); }))
if (!std::all_of(lvlTypes.begin(), it, isDenseLT))
return emitError() << "expected all dense lvlTypes "
"before a n_out_of_m level";
if (dimToLvl && (dimToLvl.getNumDims() != dimToLvl.getNumResults())) {

3
mlir/lib/Dialect/Tensor/IR/TensorOps.cpp
View File

@ -1061,8 +1061,7 @@ void DimOp::getCanonicalizationPatterns(RewritePatternSet &results,
void EmptyOp::build(OpBuilder &builder, OperationState &result,
ArrayRef<int64_t> staticShape, Type elementType,
Attribute encoding) {
assert(all_of(staticShape,
[](int64_t sz) { return !ShapedType::isDynamic(sz); }) &&
assert(none_of(staticShape, ShapedType::isDynamic) &&
"expected only static sizes");
build(builder, result, staticShape, elementType, ValueRange{}, encoding);
}