Mirror/aicodix___code
1
0
Fork 0
mirror of https://github.com/aicodix/code.git synced 2026-04-27 22:35:44 +00:00
Code Issues Projects Releases Packages Wiki Activity

added SIMD acceleration for GF(2^8) and GF(2^16)

Browse source
This commit is contained in:
Ahmet Inan 2023-05-08 12:34:39 +02:00
commit 8370e6a51c
2 changed files with 236 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

20
tests/crs_regression_test.cc
View file

@ -4,6 +4,7 @@ Regression Test for the Cauchy Reed Solomon Encoder and Decoder
Copyright 2023 Ahmet Inan <inan@aicodix.de>
*/
#include <cstdlib>
#include <cassert>
#include <chrono>
#include <random>
@ -15,6 +16,11 @@ Copyright 2023 Ahmet Inan <inan@aicodix.de>
template <typename GF>
void crs_test(int trials)
{
#ifdef __AVX2__
int SIMD = 32;
#else
int SIMD = 16;
#endif
CODE::CauchyReedSolomonErasureCoding<GF> crs;
std::random_device rd;
std::default_random_engine generator(rd());
@ -25,12 +31,13 @@ void crs_test(int trials)
while (--trials) {
int block_count = rnd_cnt();
int numbers_total = GF::Q - block_count;
int block_bytes = rnd_len() * sizeof(typename GF::value_type);
int block_bytes = rnd_len() * sizeof(typename GF::value_type) * SIMD;
int data_bytes = block_count * block_bytes;
uint8_t *orig = new uint8_t[data_bytes];
uint8_t *orig = reinterpret_cast<uint8_t *>(std::aligned_alloc(SIMD, data_bytes));
uint8_t *data = reinterpret_cast<uint8_t *>(std::aligned_alloc(SIMD, data_bytes));
uint8_t *blocks = reinterpret_cast<uint8_t *>(std::aligned_alloc(SIMD, data_bytes));
for (int i = 0; i < data_bytes; ++i)
orig[i] = rnd_dat();
uint8_t *blocks = new uint8_t[data_bytes];
auto numbers = new typename GF::value_type[numbers_total];
for (int i = 0; i < numbers_total; ++i)
numbers[i] = i;
@ -44,7 +51,6 @@ void crs_test(int trials)
auto enc_end = std::chrono::system_clock::now();
auto enc_usec = std::chrono::duration_cast<std::chrono::microseconds>(enc_end - enc_start);
double enc_mbs = double(data_bytes) / enc_usec.count();
uint8_t *data = new uint8_t[data_bytes];
auto dec_start = std::chrono::system_clock::now();
for (int i = 0; i < block_count; ++i)
crs.decode(data + block_bytes * i, blocks, numbers, i, block_bytes, block_count);
@ -55,9 +61,9 @@ void crs_test(int trials)
for (int i = 0; i < data_bytes; ++i)
assert(data[i] == orig[i]);
delete[] numbers;
delete[] blocks;
delete[] orig;
delete[] data;
std::free(blocks);
std::free(orig);
std::free(data);
}
}