From 2b6ea95d16e09b53a8bca44b135e363a5912960d Mon Sep 17 00:00:00 2001
From: Kyle Bayes <kylebayes@google.com>
Date: Wed, 14 Jun 2023 11:49:05 -0700
Subject: [PATCH] Fix dimension of res vector in sparse benchmark test.

PiperOrigin-RevId: 540336845
Change-Id: I79d4c487455a0642d57f36abe0e38eae92f212fa
---
 test/benchmark/engine_util_sparse_benchmark_test.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/benchmark/engine_util_sparse_benchmark_test.cc b/test/benchmark/engine_util_sparse_benchmark_test.cc
index ca8489a4..06687399 100644
--- a/test/benchmark/engine_util_sparse_benchmark_test.cc
+++ b/test/benchmark/engine_util_sparse_benchmark_test.cc
@@ -357,7 +357,7 @@ static void BM_MatVecSparse(benchmark::State& state, int unroll) {
   mjMARKSTACK;
   mjtNum *Ma = mj_stackAlloc(d, m->nv);
   mjtNum *vec = mj_stackAlloc(d, m->nv);
-  mjtNum *res = mj_stackAlloc(d, m->nv);
+  mjtNum *res = mj_stackAlloc(d, d->nefc);
   mjtNum *grad = mj_stackAlloc(d, m->nv);
   mjtNum *Mgrad  = mj_stackAlloc(d, m->nv);