mirror of
https://github.com/rdolbeau/VexRiscvBPluginGenerator.git
synced 2025-04-18 18:44:42 -04:00
small sum of squared error test
This commit is contained in:
Romain Dolbeau
parent
f09395cca8
commit
2107c462bf
1 changed files with 183 additions and 0 deletions
183
pbsse.c
Normal file
183
pbsse.c
Normal file
|
|
@ -0,0 +1,183 @@
|
|||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <stdint.h>
|
||||
|
||||
typedef uint32_t MpegEncContext; /* don't care */
|
||||
typedef ssize_t ptrdiff_t;
|
||||
|
||||
#ifdef __riscv
|
||||
|
||||
#include "new_instructions_support_p.h"
|
||||
|
||||
static inline long long cpucycles_riscv(void) {
|
||||
long long result;
|
||||
#if defined(__riscv_xlen)
|
||||
#if __riscv_xlen == 64
|
||||
asm volatile("rdcycle %0" : "=r" (result));
|
||||
#elif __riscv_xlen == 32
|
||||
unsigned int l, h, h2;
|
||||
asm volatile( "start:\n"
|
||||
"rdcycleh %0\n"
|
||||
"rdcycle %1\n"
|
||||
"rdcycleh %2\n"
|
||||
"bne %0, %2, start\n"
|
||||
: "=r" (h), "=r" (l), "=r" (h2));
|
||||
|
||||
result = (((unsigned long long)h)<<32) | ((unsigned long long)l);
|
||||
#else
|
||||
#error "unknown __riscv_xlen"
|
||||
#endif
|
||||
#else // __riscv_xlen
|
||||
#error "__riscv_xlen required for RISC-V support"
|
||||
#endif // __riscv_xlen
|
||||
return result;
|
||||
}
|
||||
typedef uint32_t uint2x32_t[2];
|
||||
static inline int sse8_r5vp(MpegEncContext *v, uint8_t *pix1, uint8_t *pix2,
|
||||
ptrdiff_t stride, int h)
|
||||
{
|
||||
int s = 0, i;
|
||||
|
||||
for (i = 0; i < h; i++) {
|
||||
uint32_t p1 = ((uint32_t*)pix1)[0];
|
||||
uint32_t p2 = ((uint32_t*)pix2)[0];
|
||||
uint32_t pmin = __rv__umin8(p1, p2);
|
||||
uint32_t pmax = __rv__umax8(p1, p2);
|
||||
uint32_t pd = __rv__sub8(pmax, pmin);
|
||||
uint64_t qs = __rv__umul8(pd, pd);
|
||||
/* uint32_t psl = (uint32_t)qs; */
|
||||
/* uint32_t psh = (uint32_t)(qs >> 32); */
|
||||
/* uint32_t ds = __rv__add16(psl, psh); */
|
||||
/* s += (ds & 0xFFFF); */
|
||||
/* s += (ds >> 16); */
|
||||
|
||||
s += ((qs >> 0) & 0x000000000000FFFFull);
|
||||
s += ((qs >> 16) & 0x000000000000FFFFull);
|
||||
s += ((qs >> 32) & 0x000000000000FFFFull);
|
||||
s += ((qs >> 48) & 0x000000000000FFFFull);
|
||||
|
||||
p1 = ((uint32_t*)pix1)[1];
|
||||
p2 = ((uint32_t*)pix2)[1];
|
||||
pmin = __rv__umin8(p1, p2);
|
||||
pmax = __rv__umax8(p1, p2);
|
||||
pd = __rv__sub8(pmax, pmin);
|
||||
qs = __rv__umul8(pd, pd);
|
||||
/* psl = (uint32_t)qs; */
|
||||
/* psh = (uint32_t)(qs >> 32); */
|
||||
/* ds = __rv__add16(psl, psh); */
|
||||
/* s += (ds & 0xFFFF); */
|
||||
/* s += (ds >> 16); */
|
||||
|
||||
s += ((qs >> 0) & 0x000000000000FFFFull);
|
||||
s += ((qs >> 16) & 0x000000000000FFFFull);
|
||||
s += ((qs >> 32) & 0x000000000000FFFFull);
|
||||
s += ((qs >> 48) & 0x000000000000FFFFull);
|
||||
|
||||
pix1 += stride;
|
||||
pix2 += stride;
|
||||
}
|
||||
return s;
|
||||
}
|
||||
#endif // __riscv
|
||||
|
||||
|
||||
|
||||
/* lifted from FFmpeg */
|
||||
const uint32_t ff_square_tab[512] = {
|
||||
65536, 65025, 64516, 64009, 63504, 63001, 62500, 62001, 61504, 61009, 60516, 60025, 59536, 59049, 58564, 58081,
|
||||
57600, 57121, 56644, 56169, 55696, 55225, 54756, 54289, 53824, 53361, 52900, 52441, 51984, 51529, 51076, 50625,
|
||||
50176, 49729, 49284, 48841, 48400, 47961, 47524, 47089, 46656, 46225, 45796, 45369, 44944, 44521, 44100, 43681,
|
||||
43264, 42849, 42436, 42025, 41616, 41209, 40804, 40401, 40000, 39601, 39204, 38809, 38416, 38025, 37636, 37249,
|
||||
36864, 36481, 36100, 35721, 35344, 34969, 34596, 34225, 33856, 33489, 33124, 32761, 32400, 32041, 31684, 31329,
|
||||
30976, 30625, 30276, 29929, 29584, 29241, 28900, 28561, 28224, 27889, 27556, 27225, 26896, 26569, 26244, 25921,
|
||||
25600, 25281, 24964, 24649, 24336, 24025, 23716, 23409, 23104, 22801, 22500, 22201, 21904, 21609, 21316, 21025,
|
||||
20736, 20449, 20164, 19881, 19600, 19321, 19044, 18769, 18496, 18225, 17956, 17689, 17424, 17161, 16900, 16641,
|
||||
16384, 16129, 15876, 15625, 15376, 15129, 14884, 14641, 14400, 14161, 13924, 13689, 13456, 13225, 12996, 12769,
|
||||
12544, 12321, 12100, 11881, 11664, 11449, 11236, 11025, 10816, 10609, 10404, 10201, 10000, 9801, 9604, 9409,
|
||||
9216, 9025, 8836, 8649, 8464, 8281, 8100, 7921, 7744, 7569, 7396, 7225, 7056, 6889, 6724, 6561,
|
||||
6400, 6241, 6084, 5929, 5776, 5625, 5476, 5329, 5184, 5041, 4900, 4761, 4624, 4489, 4356, 4225,
|
||||
4096, 3969, 3844, 3721, 3600, 3481, 3364, 3249, 3136, 3025, 2916, 2809, 2704, 2601, 2500, 2401,
|
||||
2304, 2209, 2116, 2025, 1936, 1849, 1764, 1681, 1600, 1521, 1444, 1369, 1296, 1225, 1156, 1089,
|
||||
1024, 961, 900, 841, 784, 729, 676, 625, 576, 529, 484, 441, 400, 361, 324, 289,
|
||||
256, 225, 196, 169, 144, 121, 100, 81, 64, 49, 36, 25, 16, 9, 4, 1,
|
||||
0, 1, 4, 9, 16, 25, 36, 49, 64, 81, 100, 121, 144, 169, 196, 225,
|
||||
256, 289, 324, 361, 400, 441, 484, 529, 576, 625, 676, 729, 784, 841, 900, 961,
|
||||
1024, 1089, 1156, 1225, 1296, 1369, 1444, 1521, 1600, 1681, 1764, 1849, 1936, 2025, 2116, 2209,
|
||||
2304, 2401, 2500, 2601, 2704, 2809, 2916, 3025, 3136, 3249, 3364, 3481, 3600, 3721, 3844, 3969,
|
||||
4096, 4225, 4356, 4489, 4624, 4761, 4900, 5041, 5184, 5329, 5476, 5625, 5776, 5929, 6084, 6241,
|
||||
6400, 6561, 6724, 6889, 7056, 7225, 7396, 7569, 7744, 7921, 8100, 8281, 8464, 8649, 8836, 9025,
|
||||
9216, 9409, 9604, 9801, 10000, 10201, 10404, 10609, 10816, 11025, 11236, 11449, 11664, 11881, 12100, 12321,
|
||||
12544, 12769, 12996, 13225, 13456, 13689, 13924, 14161, 14400, 14641, 14884, 15129, 15376, 15625, 15876, 16129,
|
||||
16384, 16641, 16900, 17161, 17424, 17689, 17956, 18225, 18496, 18769, 19044, 19321, 19600, 19881, 20164, 20449,
|
||||
20736, 21025, 21316, 21609, 21904, 22201, 22500, 22801, 23104, 23409, 23716, 24025, 24336, 24649, 24964, 25281,
|
||||
25600, 25921, 26244, 26569, 26896, 27225, 27556, 27889, 28224, 28561, 28900, 29241, 29584, 29929, 30276, 30625,
|
||||
30976, 31329, 31684, 32041, 32400, 32761, 33124, 33489, 33856, 34225, 34596, 34969, 35344, 35721, 36100, 36481,
|
||||
36864, 37249, 37636, 38025, 38416, 38809, 39204, 39601, 40000, 40401, 40804, 41209, 41616, 42025, 42436, 42849,
|
||||
43264, 43681, 44100, 44521, 44944, 45369, 45796, 46225, 46656, 47089, 47524, 47961, 48400, 48841, 49284, 49729,
|
||||
50176, 50625, 51076, 51529, 51984, 52441, 52900, 53361, 53824, 54289, 54756, 55225, 55696, 56169, 56644, 57121,
|
||||
57600, 58081, 58564, 59049, 59536, 60025, 60516, 61009, 61504, 62001, 62500, 63001, 63504, 64009, 64516, 65025,
|
||||
};
|
||||
static int sse8_c(MpegEncContext *v, uint8_t *pix1, uint8_t *pix2,
|
||||
ptrdiff_t stride, int h)
|
||||
{
|
||||
int s = 0, i;
|
||||
const uint32_t *sq = ff_square_tab + 256;
|
||||
|
||||
for (i = 0; i < h; i++) {
|
||||
s += sq[pix1[0] - pix2[0]];
|
||||
s += sq[pix1[1] - pix2[1]];
|
||||
s += sq[pix1[2] - pix2[2]];
|
||||
s += sq[pix1[3] - pix2[3]];
|
||||
s += sq[pix1[4] - pix2[4]];
|
||||
s += sq[pix1[5] - pix2[5]];
|
||||
s += sq[pix1[6] - pix2[6]];
|
||||
s += sq[pix1[7] - pix2[7]];
|
||||
pix1 += stride;
|
||||
pix2 += stride;
|
||||
}
|
||||
return s;
|
||||
}
|
||||
|
||||
int main(int argc, char **argv) {
|
||||
unsigned int seed = 0;
|
||||
uint8_t *pix1, *pix2;
|
||||
uint32_t i, j;
|
||||
|
||||
if (argc > 1)
|
||||
seed = atol(argv[1]);
|
||||
|
||||
srandom(seed);
|
||||
|
||||
pix1 = malloc(8 * 4 * 8 * 4 * sizeof(uint8_t));
|
||||
pix2 = malloc(8 * 4 * 8 * 4 * sizeof(uint8_t));
|
||||
for (i = 0 ; i < 8 * 8 * 4 * 4 ; i ++) {
|
||||
pix1[i] = random() & 0xFF;
|
||||
pix2[i] = random() & 0xFF;
|
||||
}
|
||||
|
||||
for (i = 0 ; i < 4 ; i++) {
|
||||
for (j = 0 ; j < 4 ; j++) {
|
||||
size_t offset = i*8+j*(8*4*8);
|
||||
long long t0 = 0, t1 = 0, t2 = 0, t3 = 0, t4 = 0;
|
||||
#ifdef __riscv
|
||||
t0 = cpucycles_riscv();
|
||||
#endif
|
||||
int res = sse8_c(&i, pix1 + offset, pix2 + offset, 8 * 4, 8);
|
||||
#ifdef __riscv
|
||||
t1 = cpucycles_riscv();
|
||||
int res2 = sse8_r5vp(&i, pix1 + offset, pix2 + offset, 8 * 4, 8);
|
||||
t2 = cpucycles_riscv();
|
||||
int res3 = sse8_c(&i, pix1 + offset, pix2 + offset, 8 * 4, 8);
|
||||
t3 = cpucycles_riscv();
|
||||
int res4 = sse8_r5vp(&i, pix1 + offset, pix2 + offset, 8 * 4, 8);
|
||||
t4 = cpucycles_riscv();
|
||||
#else
|
||||
int res2 = 0, res3 = 0, res4 = 0;
|
||||
#endif
|
||||
printf("%u, %u: %d, %d / %d, %d (%llu, %llu / %llu, %llu)\n", i, j, res, res2, res3, res4, t1-t0, t2-t1, t3-t2, t4-t3);
|
||||
}
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
Loading…
Add table
Reference in a new issue