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 src_buffer, // TODO(scotttodd): Clean this up somehow, share across recursion levels? size_t dst_size = src_shape.size() - reduce_dims.size(); std::vector 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)) {