benchmarks/Core/bench_matrix_checks.cpp

// SPDX-FileCopyrightText: The Eigen Authors
// SPDX-License-Identifier: MPL-2.0

#include <benchmark/benchmark.h>
#include <Eigen/Core>

template <typename Scalar, int Operation, int FailurePosition = 0>
static void matrix_checks(benchmark::State& state) {
  static_assert(Operation == 0 || Operation == 1, "Only zero and constant checks are benchmarked");
  Eigen::Matrix<Scalar, Eigen::Dynamic, Eigen::Dynamic> a(state.range(0), state.range(0));
  a.setConstant(Operation == 0 ? Scalar(0) : Scalar(0.5));
  if (FailurePosition != 0) a(FailurePosition == 1 ? 0 : a.size() - 1) = Scalar(2);
  const auto check = [&a]() { return Operation == 0 ? a.isZero() : a.isApproxToConstant(Scalar(0.5)); };
  if (check() != (FailurePosition == 0)) {
    state.SkipWithError("incorrect fuzzy comparison");
    return;
  }
  for (auto _ : state) {
    benchmark::ClobberMemory();
    bool result = check();
    benchmark::DoNotOptimize(result);
  }
}

BENCHMARK_TEMPLATE(matrix_checks, double, 0)->Arg(3)->Arg(4)->Arg(8)->Arg(16)->Arg(32)->Arg(256)->Arg(1024)->Arg(2048);
BENCHMARK_TEMPLATE(matrix_checks, double, 1)->Arg(3)->Arg(4)->Arg(8)->Arg(16)->Arg(32)->Arg(256)->Arg(1024)->Arg(2048);
BENCHMARK_TEMPLATE(matrix_checks, float, 0)->Arg(3)->Arg(32)->Arg(256)->Arg(1024)->Arg(2048);
BENCHMARK_TEMPLATE(matrix_checks, float, 1)->Arg(3)->Arg(32)->Arg(256)->Arg(1024)->Arg(2048);

BENCHMARK_TEMPLATE(matrix_checks, double, 0, 1)->Arg(3)->Arg(32)->Arg(256);
BENCHMARK_TEMPLATE(matrix_checks, double, 0, 2)->Arg(3)->Arg(32)->Arg(256);
BENCHMARK_TEMPLATE(matrix_checks, double, 1, 1)->Arg(3)->Arg(32)->Arg(256);
BENCHMARK_TEMPLATE(matrix_checks, double, 1, 2)->Arg(3)->Arg(32)->Arg(256);