c++ - 在 ARM NEON 中的数组边界上加载 vector

Question

我尝试使用 NEON 内在函数来优化 ARM 的一些图像处理算法。 对于某些过滤器，它需要加载点附近的元素。 例如，要以像素 p[x] 为单位过滤图像，我需要加载 p[x - 1]、p[x] 和 p[x + 1]。 如果 x=0，则加载 p[0]、p[0] 和 p[1] 。如果 x=width-1，则加载 p[width-2]、p[width-1] 和 p[宽度-1]。

所以如果我有一个 vector :

uint8x16_t a = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15};

如何从中获得以下 vector :

uint8x16_t b = {0, 0, 1, 2, 3, 4, 5, 6, 7,  8,  9, 10, 11, 12, 13, 14};
uint8x16_t c = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 15};

Answer 1

我认为以下功能对您的情况有用:

template <size_t count> inline uint8x16_t LoadBeforeFirst(uint8x16_t first)
{
    return vextq_u8(vextq_u8(first, first, count), first, 16 - count);
}

template <size_t count> inline uint8x16_t LoadAfterLast(uint8x16_t last)
{
    return vextq_u8(last, vextq_u8(last, last, 16 - count), count);
}