2017-10-17 12:17:11 +00:00
|
|
|
/*!
|
|
|
|
* \file fft_speed_test.cc
|
|
|
|
* \brief This file implements timing tests for the Armadillo
|
|
|
|
* and GNU Radio FFT implementations
|
|
|
|
* \author Antonio Ramos, 2017. antonio.ramos(at)cttc.es
|
|
|
|
*
|
|
|
|
*
|
|
|
|
* -------------------------------------------------------------------------
|
|
|
|
*
|
2019-07-26 10:38:20 +00:00
|
|
|
* Copyright (C) 2010-2019 (see AUTHORS file for a list of contributors)
|
2017-10-17 12:17:11 +00:00
|
|
|
*
|
|
|
|
* GNSS-SDR is a software defined Global Navigation
|
|
|
|
* Satellite Systems receiver
|
|
|
|
*
|
|
|
|
* This file is part of GNSS-SDR.
|
|
|
|
*
|
2020-02-08 00:20:02 +00:00
|
|
|
* SPDX-License-Identifier: GPL-3.0-or-later
|
2017-10-17 12:17:11 +00:00
|
|
|
*
|
|
|
|
* -------------------------------------------------------------------------
|
|
|
|
*/
|
|
|
|
|
2020-06-18 09:49:28 +00:00
|
|
|
#include "gnss_sdr_make_unique.h"
|
2018-12-09 21:00:09 +00:00
|
|
|
#include <armadillo>
|
|
|
|
#include <gnuradio/fft/fft.h>
|
2017-10-17 12:17:11 +00:00
|
|
|
#include <chrono>
|
|
|
|
#include <memory>
|
|
|
|
|
2017-10-18 10:30:42 +00:00
|
|
|
DEFINE_int32(fft_speed_iterations_test, 100, "Number of averaged iterations in FFT length timing test");
|
2017-10-17 12:17:11 +00:00
|
|
|
|
|
|
|
TEST(FFTSpeedTest, ArmadilloVSGNURadioExecutionTime)
|
|
|
|
{
|
|
|
|
unsigned int d_fft_size;
|
|
|
|
std::chrono::time_point<std::chrono::system_clock> start, end;
|
|
|
|
std::chrono::duration<double> elapsed_seconds;
|
|
|
|
|
2018-03-03 01:03:39 +00:00
|
|
|
unsigned int fft_sizes[19] = {16, 25, 32, 45, 64, 95, 128, 195, 256, 325, 512, 785, 1024, 1503, 2048, 3127, 4096, 6349, 8192};
|
2017-10-17 12:17:11 +00:00
|
|
|
double d_execution_time;
|
|
|
|
EXPECT_NO_THROW(
|
2019-02-11 14:33:24 +00:00
|
|
|
for (unsigned int fft_size
|
|
|
|
: fft_sizes) {
|
|
|
|
d_fft_size = fft_size;
|
2020-06-18 09:49:28 +00:00
|
|
|
auto d_gr_fft = std::make_unique<gr::fft::fft_complex>(d_fft_size, true);
|
2018-03-03 01:03:39 +00:00
|
|
|
arma::arma_rng::set_seed_random();
|
|
|
|
arma::cx_fvec d_arma_fft = arma::cx_fvec(d_fft_size).randn() + gr_complex(0.0, 1.0) * arma::cx_fvec(d_fft_size).randn();
|
|
|
|
arma::cx_fvec d_arma_fft_result(d_fft_size);
|
|
|
|
memcpy(d_gr_fft->get_inbuf(), d_arma_fft.memptr(), sizeof(gr_complex) * d_fft_size);
|
|
|
|
|
|
|
|
start = std::chrono::system_clock::now();
|
|
|
|
for (int k = 0; k < FLAGS_fft_speed_iterations_test; k++)
|
2017-10-17 12:17:11 +00:00
|
|
|
{
|
2018-03-03 01:03:39 +00:00
|
|
|
d_gr_fft->execute();
|
|
|
|
}
|
|
|
|
end = std::chrono::system_clock::now();
|
|
|
|
elapsed_seconds = end - start;
|
|
|
|
d_execution_time = elapsed_seconds.count() / static_cast<double>(FLAGS_fft_speed_iterations_test);
|
|
|
|
std::cout << "GNU Radio FFT execution time for length = " << d_fft_size << " : " << d_execution_time * 1e6 << " [us]" << std::endl;
|
2017-10-18 09:04:17 +00:00
|
|
|
|
2018-03-03 01:03:39 +00:00
|
|
|
start = std::chrono::system_clock::now();
|
|
|
|
for (int k = 0; k < FLAGS_fft_speed_iterations_test; k++)
|
|
|
|
{
|
|
|
|
d_arma_fft_result = arma::fft(d_arma_fft);
|
2017-10-17 12:17:11 +00:00
|
|
|
}
|
2018-03-03 01:03:39 +00:00
|
|
|
end = std::chrono::system_clock::now();
|
|
|
|
elapsed_seconds = end - start;
|
|
|
|
d_execution_time = elapsed_seconds.count() / static_cast<double>(FLAGS_fft_speed_iterations_test);
|
|
|
|
std::cout << "Armadillo FFT execution time for length = " << d_fft_size << " : " << d_execution_time * 1e6 << " [us]" << std::endl;
|
|
|
|
});
|
2017-10-17 12:17:11 +00:00
|
|
|
}
|