diff --git a/mlir/lib/Dialect/Affine/Transforms/RaiseMemrefDialect.cpp b/mlir/lib/Dialect/Affine/Transforms/RaiseMemrefDialect.cpp index 491d2e03c36bc..3fc2664aefdfb 100644 --- a/mlir/lib/Dialect/Affine/Transforms/RaiseMemrefDialect.cpp +++ b/mlir/lib/Dialect/Affine/Transforms/RaiseMemrefDialect.cpp @@ -48,7 +48,7 @@ static std::optional findInListOrAdd(Value value, llvm::SmallVectorImpl &dims, function_ref isValidElement) { - Value *loopIV = std::find(dims.begin(), dims.end(), value); + Value *loopIV = llvm::find(dims, value); if (loopIV != dims.end()) { // We found an IV that already has an index, return that index. return {std::distance(dims.begin(), loopIV)}; diff --git a/mlir/lib/Dialect/Linalg/Transforms/WinogradConv2D.cpp b/mlir/lib/Dialect/Linalg/Transforms/WinogradConv2D.cpp index f1059ddf5da2c..c6ebd3a53d981 100644 --- a/mlir/lib/Dialect/Linalg/Transforms/WinogradConv2D.cpp +++ b/mlir/lib/Dialect/Linalg/Transforms/WinogradConv2D.cpp @@ -960,7 +960,7 @@ winogradConv2DHelper(RewriterBase &rewriter, linalg::Conv2DNhwcFhwcOp convOp, F_2_3, F_4_3, F_2_5}; TransformMapKeyTy key = {m, r}; - auto it = std::find(validConfigs.begin(), validConfigs.end(), key); + auto it = llvm::find(validConfigs, key); // If we cannot find the constant transformation matrix, it means we do // not support this configuration yet. if (it == validConfigs.end()) diff --git a/mlir/lib/Dialect/Vector/Utils/VectorUtils.cpp b/mlir/lib/Dialect/Vector/Utils/VectorUtils.cpp index 399e902af4062..d5dd6f2027be8 100644 --- a/mlir/lib/Dialect/Vector/Utils/VectorUtils.cpp +++ b/mlir/lib/Dialect/Vector/Utils/VectorUtils.cpp @@ -291,8 +291,7 @@ vector::createUnrollIterator(VectorType vType, int64_t targetRank) { // cannot be unrolled). auto shapeToUnroll = vType.getShape().drop_back(targetRank); auto scalableDimsToUnroll = vType.getScalableDims().drop_back(targetRank); - auto it = - std::find(scalableDimsToUnroll.begin(), scalableDimsToUnroll.end(), true); + auto it = llvm::find(scalableDimsToUnroll, true); auto firstScalableDim = it - scalableDimsToUnroll.begin(); if (firstScalableDim == 0) return {}; diff --git a/mlir/test/lib/Dialect/Linalg/TestLinalgFusionTransforms.cpp b/mlir/test/lib/Dialect/Linalg/TestLinalgFusionTransforms.cpp index 81e7eedabd5d1..aab7f30127d96 100644 --- a/mlir/test/lib/Dialect/Linalg/TestLinalgFusionTransforms.cpp +++ b/mlir/test/lib/Dialect/Linalg/TestLinalgFusionTransforms.cpp @@ -47,8 +47,7 @@ static LogicalResult fuseLinalgOpsGreedily(func::FuncOp f) { if (failed(info)) continue; auto *originalOp = info->originalProducer.getOperation(); - auto *originalOpInLinalgOpsVector = - std::find(linalgOps.begin(), linalgOps.end(), originalOp); + auto *originalOpInLinalgOpsVector = llvm::find(linalgOps, originalOp); *originalOpInLinalgOpsVector = info->fusedProducer; // Don't mark for erasure in the tensor case, let DCE handle this. changed = true;