Yuval Tassa
7c6f519879
Deletes mj_factorI_legacy and mj_solveLD_legacy from the engine and headers. Updates factorI, solveLD, and inertia benchmarks to remove legacy targets and only benchmark CSR. Rewrites engine_core_smooth_test to verify CSR solver against mj_mulM instead of legacy solver. PiperOrigin-RevId: 942507341 Change-Id: I4281decb7018cfa3e46cb446efd5fe1179f6ae1f
84 lines
2.4 KiB
C++
84 lines
2.4 KiB
C++
// Copyright 2021 DeepMind Technologies Limited
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
// A benchmark for comparing different implementations of mj_solveLD.
|
|
|
|
#include <benchmark/benchmark.h>
|
|
#include <absl/base/attributes.h>
|
|
#include <mujoco/mjdata.h>
|
|
#include <mujoco/mujoco.h>
|
|
#include "src/engine/engine_core_smooth.h"
|
|
#include "src/engine/engine_util_misc.h"
|
|
#include "test/fixture.h"
|
|
|
|
namespace mujoco {
|
|
namespace {
|
|
|
|
// number of steps to benchmark
|
|
static const int kNumBenchmarkSteps = 50;
|
|
|
|
// ----------------------------- benchmark ------------------------------------
|
|
|
|
static void BM_solveLD(benchmark::State& state, bool coil) {
|
|
static mjModel* m;
|
|
if (coil) {
|
|
m = LoadModelFromPath("plugin/elasticity/coil.xml");
|
|
} else {
|
|
m = LoadModelFromPath("humanoid/humanoid100.xml");
|
|
}
|
|
|
|
mjData* d = mj_makeData(m);
|
|
mj_forward(m, d);
|
|
|
|
// allocate input and output vectors
|
|
mj_markStack(d);
|
|
mjtNum *vec = mj_stackAllocNum(d, m->nv);
|
|
mjtNum *res = mj_stackAllocNum(d, m->nv);
|
|
|
|
// arbitrary input vector
|
|
for (int i=0; i < m->nv; i++) {
|
|
vec[i] = 0.2 + 0.3*i;
|
|
}
|
|
|
|
// benchmark
|
|
while (state.KeepRunningBatch(kNumBenchmarkSteps)) {
|
|
for (int i=0; i < kNumBenchmarkSteps; i++) {
|
|
mju_copy(res, vec, m->nv);
|
|
mj_solveLD(res, d->qLD, d->qLDiagInv, m->nv, 1,
|
|
m->M_rownnz, m->M_rowadr, m->M_colind, nullptr);
|
|
}
|
|
}
|
|
|
|
// finalize
|
|
mj_freeStack(d);
|
|
mj_deleteData(d);
|
|
mj_deleteModel(m);
|
|
state.SetItemsProcessed(state.iterations());
|
|
}
|
|
|
|
void ABSL_ATTRIBUTE_NO_TAIL_CALL BM_solveLD_COIL_CSR(benchmark::State& state) {
|
|
MujocoErrorTestGuard guard;
|
|
BM_solveLD(state, /*coil=*/true);
|
|
}
|
|
BENCHMARK(BM_solveLD_COIL_CSR);
|
|
|
|
void ABSL_ATTRIBUTE_NO_TAIL_CALL BM_solveLD_H100_CSR(benchmark::State& state) {
|
|
MujocoErrorTestGuard guard;
|
|
BM_solveLD(state, /*coil=*/false);
|
|
}
|
|
BENCHMARK(BM_solveLD_H100_CSR);
|
|
|
|
} // namespace
|
|
} // namespace mujoco
|