X86 如何在AVX向量中找到元素的索引？

我正在尝试使用AVX编写一个硬件加速的哈希表，其中每个bucket都有一个固定的大小（AVX vector size）。出现了如何通过向量实现快速搜索的问题

不完全可能解：

example target hash: 2

<1  7  8  9  2  6  3  5>  //  vector of hashes
<2  2  2  2  2  2  2  2>  //  mask vector of target hash
------------------------  //  equality comparison
<0  0  0  0 -1  0  0  0>  //  result of comparison
<0  1  2  3  4  5  6  7>  //  vector of indexes
------------------------  //  and operation
<0  0  0  0  4  0  0  0>  //  index of target hash

目标哈希示例：2
//散列向量
//目标散列的掩码向量
------------------------//相等比较
//比较结果
//索引向量
------------------------//与操作
//目标散列的索引

如何从最后一个向量中提取目标哈希的索引？

---

使用标量积的另一个（缓慢）可能的解决方案：

<1  7  8  9  2  6  3  5>  //  vector of hashes
<2  2  2  2  2  2  2  2>  //  mask vector of target hash
------------------------  //  equality comparison
<0  0  0  0 -1  0  0  0>  //  result of comparison
<0  1  2  3  4  5  6  7>  //  vector of indexes
------------------------  //  dot
            -4

//散列向量
//目标散列的掩码向量
------------------------//相等比较
//比较结果
//索引向量
------------------------//点
-4

适用于此的水平操作是MOVMSKPS，它从XMM/YMM向量中提取掩码（基本上，它从每个通道收集顶部位）。一旦你有了它，你可以做TZCNT或LZCNT来获得一个索引

例如：

#include <intrin.h>
#include <immintrin.h>

int getIndexOf(int const values[8], int target)
{
    __m256i valuesSimd = _mm256_loadu_si256((__m256i const*)values);
    __m256i targetSplatted = _mm256_set1_epi32(target);
    __m256i equalBits = _mm256_cmpeq_epi32(valuesSimd, targetSplatted);
    unsigned equalMask = _mm256_movemask_ps(_mm256_castsi256_ps(equalBits));
    int index = _tzcnt_u32(equalMask);
    return index;
}

#包括
#包括
int getIndexOf（int常量值[8]，int目标）
{
__m256i值ssimd=_mm256_loadu_si256（（_M256iconst*）值）；
__m256i targetSplatted=_mm256_set1_epi32（目标）；
__m256i等效值=_mm256_cmpeq_epi32（值simd，目标飞溅）；
无符号equalMask=_mm256_movemask_ps（_mm256_castsi256_ps（equalBits））；
整数指数=tzcnt_32（equalMask）；
收益指数；
}