From a392a39f75af586e3d3cd046a8361939277e067f Mon Sep 17 00:00:00 2001
From: Christopher Bate <cbate@nvidia.com>
Date: Mon, 6 Jun 2022 16:08:23 -0600
Subject: [PATCH] [mlir][vector] fix typo in vector unroll transform

---
 mlir/lib/Dialect/Vector/Transforms/VectorUnrollDistribute.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mlir/lib/Dialect/Vector/Transforms/VectorUnrollDistribute.cpp b/mlir/lib/Dialect/Vector/Transforms/VectorUnrollDistribute.cpp
index b6c4114b391c1..dbca588f87f81 100644
--- a/mlir/lib/Dialect/Vector/Transforms/VectorUnrollDistribute.cpp
+++ b/mlir/lib/Dialect/Vector/Transforms/VectorUnrollDistribute.cpp
@@ -78,7 +78,7 @@ class DecomposeShapeIterator {
   // dimensions based on `order`.
   SmallVector<int64_t> delinearize(int64_t index) const {
     // Traverse in for loop order (largest stride to smallest stride).
-    SmallVector<int64_t, 4> vectorOffsets(sliceStrides.size());
+    SmallVector<int64_t> vectorOffsets(sliceStrides.size());
     for (auto idx : loopOrder) {
       vectorOffsets[idx] = index / sliceStrides[idx];
       index %= sliceStrides[idx];