如何在 ARM NEON SIMD 内函数上编码 "a[i]=b[c[i]]"

Question

我正在尝试将此 C/C++ 代码转换为 SIMD 内部函数。

for(int i=0 ; i < length ; i++)
    A[i] = B[C[i]];

我可以翻译下面的代码(C/C++)

for(int i=0 ; i < length ; i++)
    A[i] = B[i];

到 SIMD 代码（使用内部函数）

for(int i=0 ; i < length-16 ; i+=16) {
    uint8x16_t v0 = vld1q_u8(A+i);
    vst1q_u8(A+i, v0);
}

我知道关键字interleaving可以解决这个问题。但是我找不到解决方案。

谢谢。

编辑
For more information

unsigned char A [32] = {0,}; // Output Array
unsigned char B [20] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20}; // An array with values to pass to A Array
unsigned int C [32] = {19,15,11,10,5,3,6,4,5,19,10,14,16,14,8,9,10,20,11,1, 0, 3, 5, 19, 20, 11, 13, 9, 30, 31, 7}; // An array with the index information of the B array.

有没有Intrinsic function可以组成下面的代码形式？

int length = 32;
For (int i = 0; i < length-8; i+=8)
{
    Uint8x8_t v_idx = vld1_u8 (C + i);
    Uint8x8_t v = func (A, v_idx); // func (uint8_t, uint32x4_t)
    vst1_u8(C+i, v);
}

会输出20, 16, 12, 11, 6, 4, 7, 5, 6, 6, 20, 11, 15, 17, 15, 9, 10, 11, 21, 12, 2, 1, 4, 6, 20, 21, 12, 14, 10, 31, 32, 8

[注]
A 和 B 是 uint8_t * 类型，因为它们是值在 0 到 255 之间的图像，而 C 是 uint32_t * 类型，因为它们由 B 索引索引。

Answer 1

由于您没有提供很多信息，因此很难确定，但 vqtbl1_u8 可能就是您要查找的内容。它仅适用于 AArch64，尽管 vtbl1_u8 在 armv7 上可用。

一个简单的例子：

int main (void) {
  uint8_t bp[] = { 1,  1,  2,  3,  5,  8, 13, 21 };
  uint8_t cp[] = { 0,  2,  4,  6,  1,  3,  5,  7 };

  uint8x8_t b = vld1_u8(bp);
  uint8x8_t c = vld1_u8(cp);

  uint8x8_t a = vtbl1_u8(b, c);
  uint8_t ap[8];
  vst1_u8(ap, a);

  for (int x = 0 ; x < 8 ; x++)
    printf("%3u ", ap[x]);
  printf("\n");

  return 0;
}

会输出1 2 5 13 1 3 8 21

如何在 ARM NEON SIMD 内函数上编码 "a[i]=b[c[i]]"

How to code "a[i]=b[c[i]]" on ARM NEON SIMD Intrinsic function

arm

simd

intrinsics

neon