NEON：将int8x16_t解包为一对int16x8，并将一对int16x8_t解压缩为int8x16_t

Question

我正在为我制作的算法的arm64实现NEON版本。

我面临的问题是：

-如何将一个int8x16解压缩为两个int16x8_t ，这意味着字节有点“ int16x8_t ”到短裤上？
-如何将这两个int16x8_t重新打包为int8x16_t ？

我尝试执行此操作的原因是对几个向量化的短裤应用操作，而不会溢出，最后将结果打包回int8x16_t 。

这是我针对此问题的SSE2实现：

SSE2开箱：

__m128i a1 = _mm_srai_epi16(_mm_unpacklo_epi8(input, input), 8);
__m128i a2 = _mm_srai_epi16(_mm_unpackhi_epi8(input, input), 8);

SSE2包装：

__m128i output = _mm_packs_epi16(a1, a2);

Answer 1

您可以使用内部函数做到这一点，例如：

#include <stdint.h>
#include <arm_neon.h>

void func(int8_t *buf) {
    int8x16_t vec = vld1q_s8(buf); // load 16x int8_t
    int16x8_t short1 = vmovl_s8(vget_low_s8(vec)); // cast the first 8x int8_t to int16_t
    int16x8_t short2 = vmovl_s8(vget_high_s8(vec)); // cast the last 8x int8_t to int16_t
    short1 = vaddq_s16(short1, short1); // Do operation on int16
    short2 = vaddq_s16(short2, short2);
    vec = vcombine_s8(vmovn_s16(short1), vmovn_s16(short2)); // Cast back to int8_t and combine the two vectors
    vst1q_s8(buf, vec); // Store
}