mirror of
https://gitlab.com/libeigen/eigen.git
synced 2026-04-10 11:34:33 +08:00
Add benchmarks for unsupported modules and extend supported benchmarks
libeigen/eigen!2179 Closes #3036 Co-authored-by: Rasmus Munk Larsen <rmlarsen@gmail.com>
This commit is contained in:
@@ -33,11 +33,20 @@ static void BM_FFT(benchmark::State& state) {
|
||||
benchmark::Counter(mflops_per_iter, benchmark::Counter::kIsIterationInvariantRate, benchmark::Counter::kIs1000);
|
||||
}
|
||||
|
||||
BENCHMARK(BM_FFT<std::complex<float>, true>)->Arg(1024)->Arg(4096);
|
||||
BENCHMARK(BM_FFT<std::complex<float>, false>)->Arg(1024)->Arg(4096);
|
||||
BENCHMARK(BM_FFT<float, true>)->Arg(1024)->Arg(4096);
|
||||
BENCHMARK(BM_FFT<float, false>)->Arg(1024)->Arg(4096);
|
||||
BENCHMARK(BM_FFT<std::complex<double>, true>)->Arg(1024)->Arg(4096);
|
||||
BENCHMARK(BM_FFT<std::complex<double>, false>)->Arg(1024)->Arg(4096);
|
||||
BENCHMARK(BM_FFT<double, true>)->Arg(1024)->Arg(4096);
|
||||
BENCHMARK(BM_FFT<double, false>)->Arg(1024)->Arg(4096);
|
||||
static void FFTSizes(::benchmark::Benchmark* b) {
|
||||
for (int n : {64, 128, 256, 512, 1024, 2048, 4096, 8192, 16384, 65536}) {
|
||||
b->Arg(n);
|
||||
}
|
||||
// Non-power-of-2 sizes.
|
||||
b->Arg(1000);
|
||||
b->Arg(5000);
|
||||
}
|
||||
|
||||
BENCHMARK(BM_FFT<std::complex<float>, true>)->Apply(FFTSizes);
|
||||
BENCHMARK(BM_FFT<std::complex<float>, false>)->Apply(FFTSizes);
|
||||
BENCHMARK(BM_FFT<float, true>)->Apply(FFTSizes);
|
||||
BENCHMARK(BM_FFT<float, false>)->Apply(FFTSizes);
|
||||
BENCHMARK(BM_FFT<std::complex<double>, true>)->Apply(FFTSizes);
|
||||
BENCHMARK(BM_FFT<std::complex<double>, false>)->Apply(FFTSizes);
|
||||
BENCHMARK(BM_FFT<double, true>)->Apply(FFTSizes);
|
||||
BENCHMARK(BM_FFT<double, false>)->Apply(FFTSizes);
|
||||
|
||||
Reference in New Issue
Block a user