mirror of
https://gitlab.com/libeigen/eigen.git
synced 2026-04-10 11:34:33 +08:00
612 lines
21 KiB
C++
612 lines
21 KiB
C++
|
|
// Standalone tool to measure ULP accuracy of Eigen's vectorized math functions
|
||
|
|
// against either MPFR (high-precision reference) or std C++ math functions.
|
||
|
|
//
|
||
|
|
// See README.md in this directory for full documentation.
|
||
|
|
//
|
||
|
|
// Usage:
|
||
|
|
// ./ulp_accuracy --func=sin --lo=0 --hi=6.2832 --threads=16
|
||
|
|
// ./ulp_accuracy --func=exp --threads=16
|
||
|
|
// ./ulp_accuracy --func=sin --ref=mpfr
|
||
|
|
// ./ulp_accuracy --func=sin --double --step=1e-6
|
||
|
|
// ./ulp_accuracy --list
|
||
|
|
//
|
||
|
|
// Build:
|
||
|
|
// cd build && cmake .. && make ulp_accuracy
|
||
|
|
|
||
|
|
#include <Eigen/Core>
|
||
|
|
#include <unsupported/Eigen/SpecialFunctions>
|
||
|
|
|
||
|
|
#ifdef EIGEN_HAS_MPFR
|
||
|
|
#include <mpfr.h>
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#include <cfloat>
|
||
|
|
#include <chrono>
|
||
|
|
#include <cmath>
|
||
|
|
#include <cstdint>
|
||
|
|
#include <cstdio>
|
||
|
|
#include <cstring>
|
||
|
|
#include <functional>
|
||
|
|
#include <limits>
|
||
|
|
#include <memory>
|
||
|
|
#include <string>
|
||
|
|
#include <thread>
|
||
|
|
#include <type_traits>
|
||
|
|
#include <vector>
|
||
|
|
|
||
|
|
#include "mpfr_reference.h"
|
||
|
|
|
||
|
|
// ============================================================================
|
||
|
|
// ULP distance computation
|
||
|
|
// ============================================================================
|
||
|
|
|
||
|
|
// Maps IEEE 754 bits to a linear integer scale where adjacent representable
|
||
|
|
// values are adjacent integers. The mapping is strictly monotonic:
|
||
|
|
// -inf -> most negative, -0.0 -> -1, +0.0 -> 0, +inf -> most positive.
|
||
|
|
static inline int64_t scalar_to_linear(float x) {
|
||
|
|
int32_t bits;
|
||
|
|
std::memcpy(&bits, &x, sizeof(bits));
|
||
|
|
if (bits < 0) {
|
||
|
|
bits = static_cast<int32_t>(INT32_MIN) - bits - 1;
|
||
|
|
}
|
||
|
|
return static_cast<int64_t>(bits);
|
||
|
|
}
|
||
|
|
|
||
|
|
static inline int64_t scalar_to_linear(double x) {
|
||
|
|
int64_t bits;
|
||
|
|
std::memcpy(&bits, &x, sizeof(bits));
|
||
|
|
if (bits < 0) {
|
||
|
|
bits = static_cast<int64_t>(INT64_MIN) - bits - 1;
|
||
|
|
}
|
||
|
|
return bits;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Returns (eigen_val - ref_val) in ULP space.
|
||
|
|
// Positive means Eigen overestimates, negative means it underestimates.
|
||
|
|
// Returns INT64_MAX for incomparable values (NaN vs number, inf mismatch).
|
||
|
|
template <typename Scalar>
|
||
|
|
static inline int64_t signed_ulp_error(Scalar eigen_val, Scalar ref_val) {
|
||
|
|
if (eigen_val == ref_val) return 0; // also handles -0.0 == +0.0
|
||
|
|
bool e_nan = std::isnan(eigen_val), r_nan = std::isnan(ref_val);
|
||
|
|
if (e_nan && r_nan) return 0;
|
||
|
|
if (e_nan || r_nan) return INT64_MAX;
|
||
|
|
if (std::isinf(eigen_val) || std::isinf(ref_val)) return INT64_MAX;
|
||
|
|
int64_t a = scalar_to_linear(eigen_val);
|
||
|
|
int64_t b = scalar_to_linear(ref_val);
|
||
|
|
// Overflow check for a - b.
|
||
|
|
if (b > 0 && a < INT64_MIN + b) return INT64_MAX;
|
||
|
|
if (b < 0 && a > INT64_MAX + b) return INT64_MAX;
|
||
|
|
return a - b;
|
||
|
|
}
|
||
|
|
|
||
|
|
// ============================================================================
|
||
|
|
// Per-thread accumulator with signed ULP histogram
|
||
|
|
// ============================================================================
|
||
|
|
|
||
|
|
template <typename Scalar>
|
||
|
|
struct alignas(128) ThreadResult {
|
||
|
|
int64_t max_abs_ulp = 0;
|
||
|
|
Scalar max_ulp_at = Scalar(0);
|
||
|
|
Scalar max_ulp_eigen = Scalar(0);
|
||
|
|
Scalar max_ulp_ref = Scalar(0);
|
||
|
|
double abs_ulp_sum = 0.0;
|
||
|
|
uint64_t count = 0;
|
||
|
|
|
||
|
|
// Signed histogram: bins for errors in [-hist_width, +hist_width],
|
||
|
|
// plus two overflow bins for < -hist_width and > +hist_width.
|
||
|
|
// Layout: [<-W] [-W] [-W+1] ... [0] ... [W-1] [W] [>W]
|
||
|
|
// Total bins = 2*hist_width + 3
|
||
|
|
int hist_width = 0;
|
||
|
|
std::vector<uint64_t> hist;
|
||
|
|
|
||
|
|
void init(int w) {
|
||
|
|
hist_width = w;
|
||
|
|
hist.assign(2 * w + 3, 0);
|
||
|
|
}
|
||
|
|
|
||
|
|
void record(int64_t signed_err, Scalar x, Scalar eigen_val, Scalar ref_val) {
|
||
|
|
int64_t abs_err = signed_err < 0 ? -signed_err : signed_err;
|
||
|
|
if (signed_err == INT64_MAX) abs_err = INT64_MAX;
|
||
|
|
|
||
|
|
if (abs_err > max_abs_ulp) {
|
||
|
|
max_abs_ulp = abs_err;
|
||
|
|
max_ulp_at = x;
|
||
|
|
max_ulp_eigen = eigen_val;
|
||
|
|
max_ulp_ref = ref_val;
|
||
|
|
}
|
||
|
|
if (abs_err != INT64_MAX) {
|
||
|
|
abs_ulp_sum += static_cast<double>(abs_err);
|
||
|
|
}
|
||
|
|
count++;
|
||
|
|
|
||
|
|
// Histogram bin.
|
||
|
|
int bin;
|
||
|
|
if (signed_err == INT64_MAX || signed_err > hist_width) {
|
||
|
|
bin = 2 * hist_width + 2; // overflow high
|
||
|
|
} else if (signed_err < -hist_width) {
|
||
|
|
bin = 0; // overflow low
|
||
|
|
} else {
|
||
|
|
bin = static_cast<int>(signed_err) + hist_width + 1;
|
||
|
|
}
|
||
|
|
hist[bin]++;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
// ============================================================================
|
||
|
|
// Function registry
|
||
|
|
// ============================================================================
|
||
|
|
|
||
|
|
template <typename Scalar>
|
||
|
|
struct FuncEntry {
|
||
|
|
using ArrayType = Eigen::Array<Scalar, Eigen::Dynamic, 1>;
|
||
|
|
using EigenEval = std::function<void(Eigen::Ref<ArrayType>, const Eigen::Ref<const ArrayType>&)>;
|
||
|
|
using StdEval = std::function<Scalar(Scalar)>;
|
||
|
|
|
||
|
|
#ifdef EIGEN_HAS_MPFR
|
||
|
|
using MpfrEval = std::function<int(mpfr_t, const mpfr_t, mpfr_rnd_t)>;
|
||
|
|
#endif
|
||
|
|
|
||
|
|
std::string name;
|
||
|
|
EigenEval eigen_eval;
|
||
|
|
StdEval std_eval;
|
||
|
|
#ifdef EIGEN_HAS_MPFR
|
||
|
|
MpfrEval mpfr_eval;
|
||
|
|
#endif
|
||
|
|
Scalar default_lo;
|
||
|
|
Scalar default_hi;
|
||
|
|
};
|
||
|
|
|
||
|
|
// std::logistic is not part of the C++ standard library.
|
||
|
|
template <typename Scalar>
|
||
|
|
static Scalar std_logistic(Scalar x) {
|
||
|
|
if (x >= 0) {
|
||
|
|
Scalar e = std::exp(-x);
|
||
|
|
return Scalar(1) / (Scalar(1) + e);
|
||
|
|
} else {
|
||
|
|
Scalar e = std::exp(x);
|
||
|
|
return e / (Scalar(1) + e);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
template <typename Scalar>
|
||
|
|
static std::vector<FuncEntry<Scalar>> build_func_table() {
|
||
|
|
using ArrayType = Eigen::Array<Scalar, Eigen::Dynamic, 1>;
|
||
|
|
std::vector<FuncEntry<Scalar>> table;
|
||
|
|
|
||
|
|
#ifdef EIGEN_HAS_MPFR
|
||
|
|
#define ADD_FUNC(fname, eigen_expr, std_expr, mpfr_fn, lo, hi) \
|
||
|
|
table.push_back({#fname, [](Eigen::Ref<ArrayType> out, const Eigen::Ref<const ArrayType>& a) { out = eigen_expr; }, \
|
||
|
|
[](Scalar x) -> Scalar { return std_expr; }, \
|
||
|
|
[](mpfr_t r, const mpfr_t o, mpfr_rnd_t d) { return mpfr_fn(r, o, d); }, lo, hi})
|
||
|
|
#else
|
||
|
|
#define ADD_FUNC(fname, eigen_expr, std_expr, mpfr_fn, lo, hi) \
|
||
|
|
table.push_back({#fname, [](Eigen::Ref<ArrayType> out, const Eigen::Ref<const ArrayType>& a) { out = eigen_expr; }, \
|
||
|
|
[](Scalar x) -> Scalar { return std_expr; }, lo, hi})
|
||
|
|
#endif
|
||
|
|
|
||
|
|
constexpr Scalar kInf = std::numeric_limits<Scalar>::infinity();
|
||
|
|
|
||
|
|
// Trigonometric
|
||
|
|
// clang-format off
|
||
|
|
ADD_FUNC(sin, a.sin(), std::sin(x), mpfr_sin, -kInf, kInf);
|
||
|
|
ADD_FUNC(cos, a.cos(), std::cos(x), mpfr_cos, -kInf, kInf);
|
||
|
|
ADD_FUNC(tan, a.tan(), std::tan(x), mpfr_tan, -kInf, kInf);
|
||
|
|
ADD_FUNC(asin, a.asin(), std::asin(x), mpfr_asin, -kInf, kInf);
|
||
|
|
ADD_FUNC(acos, a.acos(), std::acos(x), mpfr_acos, -kInf, kInf);
|
||
|
|
ADD_FUNC(atan, a.atan(), std::atan(x), mpfr_atan, -kInf, kInf);
|
||
|
|
|
||
|
|
// Hyperbolic
|
||
|
|
ADD_FUNC(sinh, a.sinh(), std::sinh(x), mpfr_sinh, -kInf, kInf);
|
||
|
|
ADD_FUNC(cosh, a.cosh(), std::cosh(x), mpfr_cosh, -kInf, kInf);
|
||
|
|
ADD_FUNC(tanh, a.tanh(), std::tanh(x), mpfr_tanh, -kInf, kInf);
|
||
|
|
ADD_FUNC(asinh, a.asinh(), std::asinh(x), mpfr_asinh, -kInf, kInf);
|
||
|
|
ADD_FUNC(acosh, a.acosh(), std::acosh(x), mpfr_acosh, -kInf, kInf);
|
||
|
|
ADD_FUNC(atanh, a.atanh(), std::atanh(x), mpfr_atanh, -kInf, kInf);
|
||
|
|
|
||
|
|
// Exponential / Logarithmic
|
||
|
|
ADD_FUNC(exp, a.exp(), std::exp(x), mpfr_exp, -kInf, kInf);
|
||
|
|
ADD_FUNC(exp2, a.exp2(), std::exp2(x), mpfr_exp2_wrap, -kInf, kInf);
|
||
|
|
ADD_FUNC(expm1, a.expm1(), std::expm1(x), mpfr_expm1, -kInf, kInf);
|
||
|
|
ADD_FUNC(log, a.log(), std::log(x), mpfr_log, -kInf, kInf);
|
||
|
|
ADD_FUNC(log1p, a.log1p(), std::log1p(x), mpfr_log1p, -kInf, kInf);
|
||
|
|
ADD_FUNC(log10, a.log10(), std::log10(x), mpfr_log10, -kInf, kInf);
|
||
|
|
ADD_FUNC(log2, a.log2(), std::log2(x), mpfr_log2_wrap, -kInf, kInf);
|
||
|
|
|
||
|
|
// Error / Gamma
|
||
|
|
ADD_FUNC(erf, a.erf(), std::erf(x), mpfr_erf, -kInf, kInf);
|
||
|
|
ADD_FUNC(erfc, a.erfc(), std::erfc(x), mpfr_erfc, -kInf, kInf);
|
||
|
|
ADD_FUNC(lgamma, a.lgamma(), std::lgamma(x), mpfr_lngamma, -kInf, kInf);
|
||
|
|
|
||
|
|
// Other
|
||
|
|
ADD_FUNC(logistic, a.logistic(), std_logistic(x), mpfr_logistic, -kInf, kInf);
|
||
|
|
ADD_FUNC(sqrt, a.sqrt(), std::sqrt(x), mpfr_sqrt, -kInf, kInf);
|
||
|
|
ADD_FUNC(cbrt, a.cbrt(), std::cbrt(x), mpfr_cbrt, -kInf, kInf);
|
||
|
|
ADD_FUNC(rsqrt, a.rsqrt(), Scalar(1)/std::sqrt(x), mpfr_rsqrt, -kInf, kInf);
|
||
|
|
// clang-format on
|
||
|
|
|
||
|
|
#undef ADD_FUNC
|
||
|
|
return table;
|
||
|
|
}
|
||
|
|
|
||
|
|
// ============================================================================
|
||
|
|
// Range iteration helpers
|
||
|
|
// ============================================================================
|
||
|
|
|
||
|
|
// Advances x toward +inf by at least 1 ULP. When step_eps > 0, additionally
|
||
|
|
// jumps by a relative factor of (1 + step_eps) to sample the range sparsely.
|
||
|
|
template <typename Scalar>
|
||
|
|
static inline Scalar advance_by_step(Scalar x, double step_eps) {
|
||
|
|
Scalar next = std::nextafter(x, std::numeric_limits<Scalar>::infinity());
|
||
|
|
if (step_eps > 0.0 && std::isfinite(next)) {
|
||
|
|
// Try to jump further by a relative amount.
|
||
|
|
Scalar jumped = next > 0 ? next * static_cast<Scalar>(1.0 + step_eps) : next / static_cast<Scalar>(1.0 + step_eps);
|
||
|
|
// Use the jump only if it actually advances further (handles denormal stalling).
|
||
|
|
if (jumped > next) next = jumped;
|
||
|
|
}
|
||
|
|
return next;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Counts the number of representable scalars in [lo, hi].
|
||
|
|
template <typename Scalar>
|
||
|
|
static uint64_t count_scalars_in_range(Scalar lo, Scalar hi) {
|
||
|
|
if (lo > hi) return 0;
|
||
|
|
uint64_t lo_u = static_cast<uint64_t>(scalar_to_linear(lo));
|
||
|
|
uint64_t hi_u = static_cast<uint64_t>(scalar_to_linear(hi));
|
||
|
|
uint64_t diff = hi_u - lo_u;
|
||
|
|
return diff == UINT64_MAX ? UINT64_MAX : diff + 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Advances a scalar by n ULPs in the linear representation.
|
||
|
|
static float advance_scalar(float x, uint64_t n) {
|
||
|
|
int64_t lin = scalar_to_linear(x);
|
||
|
|
lin += static_cast<int64_t>(n);
|
||
|
|
int32_t ibits;
|
||
|
|
if (lin < 0) {
|
||
|
|
ibits = static_cast<int32_t>(INT32_MIN) - static_cast<int32_t>(lin) - 1;
|
||
|
|
} else {
|
||
|
|
ibits = static_cast<int32_t>(lin);
|
||
|
|
}
|
||
|
|
float result;
|
||
|
|
std::memcpy(&result, &ibits, sizeof(result));
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
|
||
|
|
static double advance_scalar(double x, uint64_t n) {
|
||
|
|
int64_t lin = scalar_to_linear(x);
|
||
|
|
lin += static_cast<int64_t>(n);
|
||
|
|
int64_t ibits;
|
||
|
|
if (lin < 0) {
|
||
|
|
ibits = static_cast<int64_t>(INT64_MIN) - lin - 1;
|
||
|
|
} else {
|
||
|
|
ibits = lin;
|
||
|
|
}
|
||
|
|
double result;
|
||
|
|
std::memcpy(&result, &ibits, sizeof(result));
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
|
||
|
|
// ============================================================================
|
||
|
|
// Worker thread: evaluates Eigen and reference over a subrange
|
||
|
|
// ============================================================================
|
||
|
|
|
||
|
|
template <typename Scalar>
|
||
|
|
static void worker(const FuncEntry<Scalar>& func, Scalar lo, Scalar hi, int batch_size, bool use_mpfr, double step_eps,
|
||
|
|
ThreadResult<Scalar>& result) {
|
||
|
|
using ArrayType = Eigen::Array<Scalar, Eigen::Dynamic, 1>;
|
||
|
|
ArrayType input(batch_size);
|
||
|
|
ArrayType eigen_out(batch_size);
|
||
|
|
std::vector<Scalar> ref_out(batch_size);
|
||
|
|
|
||
|
|
#ifdef EIGEN_HAS_MPFR
|
||
|
|
mpfr_t mp_in, mp_out;
|
||
|
|
if (use_mpfr) {
|
||
|
|
mpfr_init2(mp_in, 128);
|
||
|
|
mpfr_init2(mp_out, 128);
|
||
|
|
}
|
||
|
|
#else
|
||
|
|
(void)use_mpfr;
|
||
|
|
#endif
|
||
|
|
|
||
|
|
auto process_batch = [&](int n, const ArrayType& in, const ArrayType& eig) {
|
||
|
|
for (int i = 0; i < n; i++) {
|
||
|
|
#ifdef EIGEN_HAS_MPFR
|
||
|
|
if (use_mpfr) {
|
||
|
|
mpfr_set_scalar<Scalar>(mp_in, in[i], MPFR_RNDN);
|
||
|
|
func.mpfr_eval(mp_out, mp_in, MPFR_RNDN);
|
||
|
|
ref_out[i] = mpfr_get_scalar<Scalar>(mp_out, MPFR_RNDN);
|
||
|
|
} else
|
||
|
|
#endif
|
||
|
|
{
|
||
|
|
ref_out[i] = func.std_eval(in[i]);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
for (int i = 0; i < n; i++) {
|
||
|
|
int64_t err = signed_ulp_error(eig[i], ref_out[i]);
|
||
|
|
result.record(err, in[i], eig[i], ref_out[i]);
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
int idx = 0;
|
||
|
|
Scalar x = lo;
|
||
|
|
for (;;) {
|
||
|
|
input[idx] = x;
|
||
|
|
idx++;
|
||
|
|
|
||
|
|
if (idx == batch_size) {
|
||
|
|
func.eigen_eval(eigen_out, input);
|
||
|
|
process_batch(batch_size, input, eigen_out);
|
||
|
|
idx = 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (x >= hi) break;
|
||
|
|
Scalar next = advance_by_step(x, step_eps);
|
||
|
|
x = (next > hi) ? hi : next;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Process remaining partial batch.
|
||
|
|
if (idx > 0) {
|
||
|
|
auto partial_in = input.head(idx);
|
||
|
|
auto partial_eigen = eigen_out.head(idx);
|
||
|
|
func.eigen_eval(partial_eigen, partial_in);
|
||
|
|
process_batch(idx, input, eigen_out);
|
||
|
|
}
|
||
|
|
|
||
|
|
#ifdef EIGEN_HAS_MPFR
|
||
|
|
if (use_mpfr) {
|
||
|
|
mpfr_clear(mp_in);
|
||
|
|
mpfr_clear(mp_out);
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
}
|
||
|
|
|
||
|
|
// ============================================================================
|
||
|
|
// Test driver: splits range across threads and prints results
|
||
|
|
// ============================================================================
|
||
|
|
|
||
|
|
struct Options {
|
||
|
|
std::string func_name;
|
||
|
|
double lo = std::numeric_limits<double>::quiet_NaN();
|
||
|
|
double hi = std::numeric_limits<double>::quiet_NaN();
|
||
|
|
int num_threads;
|
||
|
|
int batch_size = 4096;
|
||
|
|
int hist_width = 10;
|
||
|
|
bool use_mpfr = false;
|
||
|
|
bool use_double = false;
|
||
|
|
double step_eps = 0.0;
|
||
|
|
bool list_funcs = false;
|
||
|
|
};
|
||
|
|
|
||
|
|
template <typename Scalar>
|
||
|
|
static int run_test(const Options& opts) {
|
||
|
|
const int kDigits = std::is_same<Scalar, float>::value ? 9 : 17;
|
||
|
|
const char* kTypeName = std::is_same<Scalar, float>::value ? "float" : "double";
|
||
|
|
|
||
|
|
auto table = build_func_table<Scalar>();
|
||
|
|
|
||
|
|
if (opts.list_funcs) {
|
||
|
|
std::printf("Available functions:\n");
|
||
|
|
for (const auto& f : table) {
|
||
|
|
std::printf(" %s\n", f.name.c_str());
|
||
|
|
}
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Look up the requested function.
|
||
|
|
const FuncEntry<Scalar>* entry = nullptr;
|
||
|
|
for (const auto& f : table) {
|
||
|
|
if (f.name == opts.func_name) {
|
||
|
|
entry = &f;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if (!entry) {
|
||
|
|
std::fprintf(stderr, "Error: unknown function '%s' (use --list to see available functions)\n",
|
||
|
|
opts.func_name.c_str());
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
Scalar lo = std::isnan(opts.lo) ? entry->default_lo : static_cast<Scalar>(opts.lo);
|
||
|
|
Scalar hi = std::isnan(opts.hi) ? entry->default_hi : static_cast<Scalar>(opts.hi);
|
||
|
|
uint64_t total_scalars = count_scalars_in_range(lo, hi);
|
||
|
|
int num_threads = opts.num_threads;
|
||
|
|
|
||
|
|
// Print test configuration.
|
||
|
|
std::printf("Function: %s (%s)\n", opts.func_name.c_str(), kTypeName);
|
||
|
|
std::printf("Range: [%.*g, %.*g]\n", kDigits, double(lo), kDigits, double(hi));
|
||
|
|
if (opts.step_eps > 0.0) {
|
||
|
|
std::printf("Sampling step: (1 + %g) * nextafter(x)\n", opts.step_eps);
|
||
|
|
} else {
|
||
|
|
std::printf("Representable values in range: %lu\n", static_cast<unsigned long>(total_scalars));
|
||
|
|
}
|
||
|
|
std::printf("Reference: %s\n", opts.use_mpfr ? "MPFR (128-bit)" : "std C++ math");
|
||
|
|
std::printf("Threads: %d\n", num_threads);
|
||
|
|
std::printf("Batch size: %d\n", opts.batch_size);
|
||
|
|
std::printf("\n");
|
||
|
|
std::fflush(stdout);
|
||
|
|
|
||
|
|
// Split range across threads.
|
||
|
|
if (total_scalars > 0 && static_cast<uint64_t>(num_threads) > total_scalars) {
|
||
|
|
num_threads = static_cast<int>(total_scalars);
|
||
|
|
}
|
||
|
|
if (num_threads < 1) num_threads = 1;
|
||
|
|
|
||
|
|
// Heap-allocate each ThreadResult separately to avoid false sharing.
|
||
|
|
std::vector<std::unique_ptr<ThreadResult<Scalar>>> results;
|
||
|
|
results.reserve(num_threads);
|
||
|
|
for (int t = 0; t < num_threads; t++) {
|
||
|
|
results.push_back(std::make_unique<ThreadResult<Scalar>>());
|
||
|
|
results.back()->init(opts.hist_width);
|
||
|
|
}
|
||
|
|
|
||
|
|
std::vector<std::thread> threads;
|
||
|
|
uint64_t scalars_per_thread = total_scalars / num_threads;
|
||
|
|
Scalar chunk_lo = lo;
|
||
|
|
|
||
|
|
auto start_time = std::chrono::steady_clock::now();
|
||
|
|
|
||
|
|
for (int t = 0; t < num_threads; t++) {
|
||
|
|
Scalar chunk_hi;
|
||
|
|
if (t == num_threads - 1) {
|
||
|
|
chunk_hi = hi;
|
||
|
|
} else {
|
||
|
|
chunk_hi = advance_scalar(chunk_lo, scalars_per_thread - 1);
|
||
|
|
}
|
||
|
|
threads.emplace_back(worker<Scalar>, std::cref(*entry), chunk_lo, chunk_hi, opts.batch_size, opts.use_mpfr,
|
||
|
|
opts.step_eps, std::ref(*results[t]));
|
||
|
|
chunk_lo = std::nextafter(chunk_hi, std::numeric_limits<Scalar>::infinity());
|
||
|
|
}
|
||
|
|
|
||
|
|
for (auto& t : threads) t.join();
|
||
|
|
auto end_time = std::chrono::steady_clock::now();
|
||
|
|
double elapsed = std::chrono::duration<double>(end_time - start_time).count();
|
||
|
|
|
||
|
|
// Reduce per-thread results.
|
||
|
|
ThreadResult<Scalar> global;
|
||
|
|
global.init(opts.hist_width);
|
||
|
|
for (int t = 0; t < num_threads; t++) {
|
||
|
|
const auto& r = *results[t];
|
||
|
|
if (r.max_abs_ulp > global.max_abs_ulp) {
|
||
|
|
global.max_abs_ulp = r.max_abs_ulp;
|
||
|
|
global.max_ulp_at = r.max_ulp_at;
|
||
|
|
global.max_ulp_eigen = r.max_ulp_eigen;
|
||
|
|
global.max_ulp_ref = r.max_ulp_ref;
|
||
|
|
}
|
||
|
|
global.abs_ulp_sum += r.abs_ulp_sum;
|
||
|
|
global.count += r.count;
|
||
|
|
for (size_t b = 0; b < global.hist.size(); b++) {
|
||
|
|
global.hist[b] += r.hist[b];
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
double mean_ulp = global.count > 0 ? global.abs_ulp_sum / global.count : 0.0;
|
||
|
|
|
||
|
|
// Print results.
|
||
|
|
std::printf("Results:\n");
|
||
|
|
std::printf(" Values tested: %lu\n", static_cast<unsigned long>(global.count));
|
||
|
|
std::printf(" Time: %.2f seconds (%.1f Mvalues/s)\n", elapsed, global.count / elapsed / 1e6);
|
||
|
|
if (global.max_abs_ulp == INT64_MAX) {
|
||
|
|
std::printf(" Max |ULP error|: inf\n");
|
||
|
|
} else {
|
||
|
|
std::printf(" Max |ULP error|: %ld\n", static_cast<long>(global.max_abs_ulp));
|
||
|
|
}
|
||
|
|
std::printf(" at x = %.*g (Eigen=%.*g, ref=%.*g)\n", kDigits, double(global.max_ulp_at), kDigits,
|
||
|
|
double(global.max_ulp_eigen), kDigits, double(global.max_ulp_ref));
|
||
|
|
std::printf(" Mean |ULP error|: %.4f\n", mean_ulp);
|
||
|
|
std::printf("\n");
|
||
|
|
|
||
|
|
// Print signed error histogram.
|
||
|
|
std::printf("Signed ULP error histogram [-%d, +%d]:\n", opts.hist_width, opts.hist_width);
|
||
|
|
int nbins = 2 * opts.hist_width + 3;
|
||
|
|
for (int b = 0; b < nbins; b++) {
|
||
|
|
if (global.hist[b] == 0) continue;
|
||
|
|
double pct = 100.0 * global.hist[b] / global.count;
|
||
|
|
if (b == 0) {
|
||
|
|
std::printf(" <%-4d: %12lu (%7.3f%%)\n", -opts.hist_width, static_cast<unsigned long>(global.hist[b]), pct);
|
||
|
|
} else if (b == nbins - 1) {
|
||
|
|
std::printf(" >%-4d: %12lu (%7.3f%%)\n", opts.hist_width, static_cast<unsigned long>(global.hist[b]), pct);
|
||
|
|
} else {
|
||
|
|
int err = b - opts.hist_width - 1;
|
||
|
|
std::printf(" %-5d: %12lu (%7.3f%%)\n", err, static_cast<unsigned long>(global.hist[b]), pct);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
// ============================================================================
|
||
|
|
// Command-line parsing
|
||
|
|
// ============================================================================
|
||
|
|
|
||
|
|
static void print_usage() {
|
||
|
|
std::printf(
|
||
|
|
"Usage: ulp_accuracy [options]\n"
|
||
|
|
" --func=NAME Function to test (required unless --list)\n"
|
||
|
|
" --lo=VAL Start of range (default: -inf)\n"
|
||
|
|
" --hi=VAL End of range (default: +inf)\n"
|
||
|
|
" --double Test double precision (default: float)\n"
|
||
|
|
" --step=EPS Sampling step: advance by (1+EPS)*nextafter(x)\n"
|
||
|
|
" (default: 0 = exhaustive; useful for double, e.g. 1e-6)\n"
|
||
|
|
" --threads=N Number of threads (default: all cores)\n"
|
||
|
|
" --batch=N Batch size for Eigen eval (default: 4096)\n"
|
||
|
|
" --ref=MODE Reference: 'std' (default) or 'mpfr'\n"
|
||
|
|
" --hist_width=N Histogram half-width in ULPs (default: 10)\n"
|
||
|
|
" --list List available functions\n");
|
||
|
|
}
|
||
|
|
|
||
|
|
int main(int argc, char* argv[]) {
|
||
|
|
Options opts;
|
||
|
|
opts.num_threads = static_cast<int>(std::thread::hardware_concurrency());
|
||
|
|
if (opts.num_threads == 0) opts.num_threads = 4;
|
||
|
|
std::string ref_mode;
|
||
|
|
|
||
|
|
for (int i = 1; i < argc; i++) {
|
||
|
|
std::string arg = argv[i];
|
||
|
|
if (arg.substr(0, 7) == "--func=") {
|
||
|
|
opts.func_name = arg.substr(7);
|
||
|
|
} else if (arg.substr(0, 5) == "--lo=") {
|
||
|
|
std::string val = arg.substr(5);
|
||
|
|
if (val == "inf")
|
||
|
|
opts.lo = std::numeric_limits<double>::infinity();
|
||
|
|
else if (val == "-inf")
|
||
|
|
opts.lo = -std::numeric_limits<double>::infinity();
|
||
|
|
else
|
||
|
|
opts.lo = std::stod(val);
|
||
|
|
} else if (arg.substr(0, 5) == "--hi=") {
|
||
|
|
std::string val = arg.substr(5);
|
||
|
|
if (val == "inf")
|
||
|
|
opts.hi = std::numeric_limits<double>::infinity();
|
||
|
|
else if (val == "-inf")
|
||
|
|
opts.hi = -std::numeric_limits<double>::infinity();
|
||
|
|
else
|
||
|
|
opts.hi = std::stod(val);
|
||
|
|
} else if (arg.substr(0, 10) == "--threads=") {
|
||
|
|
opts.num_threads = std::stoi(arg.substr(10));
|
||
|
|
} else if (arg.substr(0, 8) == "--batch=") {
|
||
|
|
opts.batch_size = std::stoi(arg.substr(8));
|
||
|
|
} else if (arg.substr(0, 6) == "--ref=") {
|
||
|
|
ref_mode = arg.substr(6);
|
||
|
|
} else if (arg.substr(0, 13) == "--hist_width=") {
|
||
|
|
opts.hist_width = std::stoi(arg.substr(13));
|
||
|
|
} else if (arg.substr(0, 7) == "--step=") {
|
||
|
|
opts.step_eps = std::stod(arg.substr(7));
|
||
|
|
} else if (arg == "--double") {
|
||
|
|
opts.use_double = true;
|
||
|
|
} else if (arg == "--list") {
|
||
|
|
opts.list_funcs = true;
|
||
|
|
} else if (arg == "--help" || arg == "-h") {
|
||
|
|
print_usage();
|
||
|
|
return 0;
|
||
|
|
} else {
|
||
|
|
std::fprintf(stderr, "Unknown option: %s\n", arg.c_str());
|
||
|
|
print_usage();
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Determine reference mode (default: std).
|
||
|
|
if (ref_mode.empty() || ref_mode == "std") {
|
||
|
|
opts.use_mpfr = false;
|
||
|
|
} else if (ref_mode == "mpfr") {
|
||
|
|
#ifdef EIGEN_HAS_MPFR
|
||
|
|
opts.use_mpfr = true;
|
||
|
|
#else
|
||
|
|
std::fprintf(stderr, "Error: MPFR support not compiled in. Use --ref=std or rebuild with MPFR.\n");
|
||
|
|
return 1;
|
||
|
|
#endif
|
||
|
|
} else {
|
||
|
|
std::fprintf(stderr, "Error: --ref must be 'std' or 'mpfr'\n");
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!opts.list_funcs && opts.func_name.empty()) {
|
||
|
|
std::fprintf(stderr, "Error: --func=NAME is required (use --list to see available functions)\n");
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (opts.use_double) {
|
||
|
|
return run_test<double>(opts);
|
||
|
|
} else {
|
||
|
|
return run_test<float>(opts);
|
||
|
|
}
|
||
|
|
}
|