1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
/* lsxintrin_ext.h
* Copyright (C) 2025 Vladislav Shchapov <vladislav@shchapov.ru>
* For conditions of distribution and use, see copyright notice in zlib.h
*/
#ifndef LSXINTRIN_EXT_H
#define LSXINTRIN_EXT_H
#include <lsxintrin.h>
static inline __m128i lsx_sad_bu(__m128i a, __m128i b) {
__m128i tmp = __lsx_vabsd_bu(a, b);
tmp = __lsx_vhaddw_hu_bu(tmp, tmp);
tmp = __lsx_vhaddw_wu_hu(tmp, tmp);
return __lsx_vhaddw_du_wu(tmp, tmp);
}
static inline int lsx_movemask_b(__m128i v) {
return __lsx_vpickve2gr_w(__lsx_vmskltz_b(v), 0);
}
static inline __m128i lsx_shuffle_b(__m128i a, __m128i b) {
/* most significant bit is set - negative 8-bit integer */
__m128i msb_mask = __lsx_vslti_b(b, 0);
/* shuffle, clear msb in indices vector b */
__m128i dst = __lsx_vshuf_b(a, a, __lsx_vandi_b(b, 0xF));
/* invert and apply mask - clear dst-element if b-msb is set */
return __lsx_vand_v(dst, __lsx_vnor_v(msb_mask, msb_mask));
}
#endif // include guard LSXINTRIN_EXT_H
|
