Update src/cunumeric/unary/scalar_unary_red_template.inl

Co-authored-by: Manolis Papadakis <[email protected]>
nv-legate · Nov 17, 2023 · 4014239 · 4014239
1 parent 6896b4e
commit 4014239
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/cunumeric/unary/scalar_unary_red_template.inl b/src/cunumeric/unary/scalar_unary_red_template.inl
@@ -165,7 +165,7 @@ static void scalar_unary_red_template(TaskContext& context)
   bool has_where   = scalars[2].value<bool>();
   size_t start_idx = has_where ? 2 : 1;
   std::vector<Store> extra_args;
-  extra_args.reserve(inputs.size());
+  extra_args.reserve(inputs.size() - start_idx);
   for (size_t idx = start_idx; idx < inputs.size(); ++idx)
     extra_args.emplace_back(std::move(inputs[idx]));
   // If the RHS was a scalar, use (1,) as the shape