diff --git a/mlir/lib/Dialect/Tensor/IR/TensorOps.cpp b/mlir/lib/Dialect/Tensor/IR/TensorOps.cpp index 5029ed4aa0387..425ebbe7f5601 100644 --- a/mlir/lib/Dialect/Tensor/IR/TensorOps.cpp +++ b/mlir/lib/Dialect/Tensor/IR/TensorOps.cpp @@ -3880,7 +3880,7 @@ static SmallVector getPackOpResultTypeShape( ArrayRef sourceShape, ArrayRef innerTileSizes, ArrayRef innerDimsPos, ArrayRef outerDimsPerm) { SmallVector resultShape = llvm::to_vector(sourceShape); - for (auto tiledDim : llvm::enumerate(innerDimsPos)) { + for (auto tiledDim : llvm::enumerate(llvm::to_vector(innerDimsPos))) { if (ShapedType::isDynamic(resultShape[tiledDim.value()])) continue; if (ShapedType::isDynamic(innerTileSizes[tiledDim.index()])) { @@ -3909,7 +3909,7 @@ SmallVector PackOp::getResultShape( AffineExpr s0, s1; bindSymbols(builder.getContext(), s0, s1); AffineExpr ceilDivExpr = s0.ceilDiv(s1); - for (auto tiledDim : llvm::enumerate(innerDimsPos)) { + for (auto tiledDim : llvm::enumerate(llvm::to_vector(innerDimsPos))) { resultDims[tiledDim.value()] = affine::makeComposedFoldedAffineApply( builder, loc, ceilDivExpr, {resultDims[tiledDim.value()], innerTileSizes[tiledDim.index()]});