diff --git a/bench/subgraph/static-reduce.cc b/bench/subgraph/static-reduce.cc index b2d91b90def..829e96e3cd2 100644 --- a/bench/subgraph/static-reduce.cc +++ b/bench/subgraph/static-reduce.cc @@ -77,12 +77,12 @@ static void FP32Reduce(benchmark::State& state) { const size_t d1 = state.range(2); const size_t d2 = state.range(3); const size_t norm_mask = state.range(4); - std::vector dims = {d0, d1, d2}; + std::vector dims = {1000, d0, d1, d2}; std::vector axes; axes.reserve(dims.size()); for (size_t i = 0; i < dims.size(); ++i) { if ((norm_mask & (1 << i)) != 0) { - axes.push_back(i); + axes.push_back(i + 1); } }