diff --git a/mlir/lib/Dialect/Vector/Transforms/VectorUnroll.cpp b/mlir/lib/Dialect/Vector/Transforms/VectorUnroll.cpp index 04c38f9f7b2e3..cb348b28c876a 100644 --- a/mlir/lib/Dialect/Vector/Transforms/VectorUnroll.cpp +++ b/mlir/lib/Dialect/Vector/Transforms/VectorUnroll.cpp @@ -13,12 +13,11 @@ #include "mlir/Dialect/Affine/IR/AffineOps.h" #include "mlir/Dialect/Utils/IndexingUtils.h" #include "mlir/Dialect/Vector/Transforms/VectorTransforms.h" -#include "mlir/IR/ImplicitLocOpBuilder.h" #include "mlir/Interfaces/VectorInterfaces.h" #include "llvm/ADT/MapVector.h" #include "llvm/ADT/STLExtras.h" #include "llvm/Support/Debug.h" -#include +#include "llvm/Support/InterleavedRange.h" #include #define DEBUG_TYPE "vector-unroll" @@ -28,7 +27,7 @@ using namespace mlir; using namespace mlir::vector; -/// Compute the indices of the slice `index` for a tranfer op. +/// Compute the indices of the slice `index` for a transfer op. static SmallVector sliceTransferIndices(ArrayRef elementOffsets, ArrayRef indices, AffineMap permutationMap, @@ -88,17 +87,14 @@ getTargetShape(const vector::UnrollVectorOptions &options, Operation *op) { LDBG("--could not get shape of op " << *op << " -> BAIL"); return std::nullopt; } - LLVM_DEBUG( - llvm::interleaveComma(*maybeUnrollShape, DBGS() << "--vector op shape: "); - llvm::dbgs() << "\n";); + LDBG("--vector op shape: " << llvm::interleaved(*maybeUnrollShape)); std::optional> targetShape = options.nativeShape(op); if (!targetShape) { LDBG("--no unrolling target shape defined " << *op << "-> SKIP"); return std::nullopt; } - LLVM_DEBUG(llvm::interleaveComma(*targetShape, DBGS() << "--target shape: "); - llvm::dbgs() << "\n";); + LDBG("--target shape: " << llvm::interleaved(*targetShape)); auto maybeShapeRatio = computeShapeRatio(*maybeUnrollShape, *targetShape); if (!maybeShapeRatio) { diff --git a/mlir/lib/Dialect/Vector/Utils/VectorUtils.cpp b/mlir/lib/Dialect/Vector/Utils/VectorUtils.cpp index 024e8ccb901de..16a2732b53e97 100644 --- a/mlir/lib/Dialect/Vector/Utils/VectorUtils.cpp +++ b/mlir/lib/Dialect/Vector/Utils/VectorUtils.cpp @@ -27,7 +27,7 @@ #include "mlir/Support/LLVM.h" #include "llvm/ADT/DenseSet.h" -#include "llvm/ADT/SetVector.h" +#include "llvm/Support/InterleavedRange.h" #define DEBUG_TYPE "vector-utils" @@ -372,9 +372,7 @@ Value vector::createReadOrMaskedRead(OpBuilder &builder, Location loc, LogicalResult vector::isValidMaskedInputVector(ArrayRef shape, ArrayRef inputVectorSizes) { - LDBG("Iteration space static sizes:"); - LLVM_DEBUG(llvm::interleaveComma(shape, llvm::dbgs())); - LLVM_DEBUG(llvm::dbgs() << "\n"); + LDBG("Iteration space static sizes:" << llvm::interleaved(shape)); if (inputVectorSizes.size() != shape.size()) { LDBG("Input vector sizes don't match the number of loops");