从`__m256i`使用值来访问有效的阵列 - SIMD

Question

比方说，例如，我有2个变量__m256i叫rows和cols，在他们里面的值是：现在

rows: 0, 2, 7, 5, 7, 2, 3, 0 
cols: 1, 2, 7, 5, 7, 2, 2, 6 

，这些值表示为8个点的x和y位置，从而，在这种情况下，我将有以下几点：

p0: [0, 1], p1: [2, 2], p2: [7, 7], p3: [5, 5] 
p4: [7, 7], p5: [2, 2], p6: [3, 2], p7: [0, 6] 

我也有一个阵列称为lut将具有的值int类型：

lut: [0, 1, 2, 3, ..., 60, 61, 62, 63] 

我想要做的，是从rows和cols变量使用这些位置值，用它访问lut阵列并创建一个新__m256i值与lut访问值。

我知道该怎么做，这将是存储rows和cols值大小为8两个int阵列，然后在同一时间从lut阵列一个读出的值，然后使用_mm256_set_epi32()创建新_m256i价值的方式。

这有效，但在我看来是非常低效的..所以我的问题是如果有办法做得更快。

请注意，这些值仅仅是一个更具体的例子，并lut并不需要有有序值或大小64

的感谢！

Answer 1

您可以构建使用avx2 gather instruction的解决方案，像这样

// index = (rows << 3) + cols; 
const __m256i index = _mm256_add_epi32(_mm256_slli_epi32(rows, 3), cols); 
// result = lut[index]; 
const __m256i result = _mm256_i32gather_epi32(lut, index, 4); 

要知道，目前CPU的收集指令有相当庞大的延迟，所以除非你可以实际使用result之前交错一些指令，这可能不是值得使用。

为了解释的4因素：scale因子在

__m256i _mm256_i32gather_epi32 (int const* base_addr, __m256i vindex, const int scale) 

被认为是实际的字节偏移量，即，用于每个索引的返回的值是：

*(const int*)((const char*) base_addr + scale*index) 

我不要知道这种行为是否有很多用例（也许这是为了能够使用1字节或2字节的条目访问LUT（以后需要一些掩码））。也许这只是允许的，因为可以缩放4，而缩放1/4或1/2不会（如果有人真的需要这个）。