forked from Firmware/Firmwares
93 lines
2.5 KiB
C++
93 lines
2.5 KiB
C++
#include "raven_spn.h"
|
|
#include "spn_regs.h"
|
|
#include "init.h"
|
|
|
|
#include <cmath> // log
|
|
|
|
using spn =spn_regs<0x90000000>;
|
|
|
|
// huge arrays of XSPN input and referance data
|
|
extern std::array<uint8_t, 50000> input_data;
|
|
extern std::array<double, 10000> ref_data;
|
|
constexpr auto ln2 = std::log(2);
|
|
|
|
bool double_equals(double a, double b, double epsilon = 0.001)
|
|
{
|
|
return std::abs(a - b) < epsilon;
|
|
}
|
|
|
|
void run_xspn(int in_addr, int out_addr) {
|
|
spn::mode_reg() = 0;
|
|
spn::input_length_reg() = 500; // each sample consists of 5 uint8 values
|
|
spn::input_addr_reg() = in_addr;
|
|
spn::output_addr_reg() = out_addr;
|
|
spn::num_of_in_beats_reg() = 40; // Number of AXI4 burst beats needed to load all input data
|
|
spn::num_of_out_beats_reg() = 64; // Number of AXI4 burst beats needed to store all result data
|
|
spn::start_reg() = 1;
|
|
}
|
|
|
|
void load_input_data(int addr, const int k=0) {
|
|
int * mem_base = (int *) addr;
|
|
int offset = k * 5;
|
|
|
|
for (size_t i = 0, j = 0; j < 625; i += 4, j++) { // load 5000 samples
|
|
*(mem_base + j) = input_data.at(offset + i);
|
|
*(mem_base + j) |= (input_data.at(offset + i + 1) << 8) & 0x0000FF00;
|
|
*(mem_base + j) |= (input_data.at(offset + i + 2) << 16) & 0x00FF0000;
|
|
*(mem_base + j) |= (input_data.at(offset + i + 3) << 24) & 0xFF000000;
|
|
}
|
|
}
|
|
|
|
bool check_results(int addr, int k) {
|
|
bool result = 0;
|
|
int step = 500;
|
|
double *res_base = (double*) (addr);
|
|
int * error_exit = (int *)0xF0000000;
|
|
|
|
for (int i = 0; i < step; i++) {
|
|
if (!double_equals(res_base[i] * ln2, ref_data.at(k + i))) {
|
|
printf("XSPN ref %d comparison FAILED\n", k + i);
|
|
*error_exit = 0x1;
|
|
result = 1;
|
|
}
|
|
}
|
|
printf("Compared samples %d - %d with the reference\n", k, k+step);
|
|
return result;
|
|
}
|
|
|
|
/*! \brief main function
|
|
*
|
|
*/
|
|
int main() {
|
|
platform_init();
|
|
int ret_val = 0;
|
|
|
|
spn::mode_reg() = 1;
|
|
spn::start_reg() = 1;
|
|
wait_for_interrupt();
|
|
spn::interrupt_reg() = 1;
|
|
printf("READOUT HW:0x%x\n", spn::readout_reg());
|
|
|
|
int in_addr = 0x800C0000;
|
|
int out_addr = 0x800A0000;
|
|
load_input_data(in_addr);
|
|
|
|
int step = 500; // number of samples to be process at once
|
|
for (int k = 0; k < 10000; k+=step) {
|
|
|
|
run_xspn(in_addr, out_addr);
|
|
printf("XSPN processes samples %d - %d\n", k, k+step);
|
|
|
|
// pre-load data for the next run while HW accelerator is running
|
|
if(k<(10000-step))
|
|
load_input_data(in_addr, k+step);
|
|
|
|
wait_for_interrupt();
|
|
spn::interrupt_reg() = 1;
|
|
|
|
ret_val = check_results(out_addr, k);
|
|
}
|
|
|
|
return ret_val;
|
|
}
|