From 2acef45fb28dd808cb541aa43c6d9fecc3e38c0f Mon Sep 17 00:00:00 2001
From: Scott Todd <scotttodd@google.com>
Date: Mon, 5 Apr 2021 13:12:38 -0700
Subject: [PATCH] Patch VMLA performance by reserve vector size before pushing
 to it.

---
 iree/modules/vmla/op_kernels_generic.h | 1 +
 1 file changed, 1 insertion(+)

diff --git a/iree/modules/vmla/op_kernels_generic.h b/iree/modules/vmla/op_kernels_generic.h
index f634da86ed7b..203f23de9c98 100644
--- a/iree/modules/vmla/op_kernels_generic.h
+++ b/iree/modules/vmla/op_kernels_generic.h
@@ -961,6 +961,7 @@ inline void ReduceDimension(absl::Span<const T> src_buffer,
     // TODO(scotttodd): Clean this up somehow, share across recursion levels?
     size_t dst_size = src_shape.size() - reduce_dims.size();
     std::vector<int> dst_indices;
+    dst_indices.reserve(src_indices.size());
     for (size_t i = 0; i < src_indices.size(); ++i) {
       if (std::find(std::begin(reduce_dims), std::end(reduce_dims), i) ==
           std::end(reduce_dims)) {