使用 SIMD 的查找表

Look-Up Table using SIMD

本文关键字：查找 SIMD 使用      更新时间：2023-10-16

我有一个大像素处理函数，我目前正在尝试使用内在函数进行优化。

作为SSE新手，我不确定如何处理涉及查找表的代码部分。

基本上，我正在尝试矢量化以下原版C++代码：

 //outside loop
const float LUT_RATIO = 1000.0F;
//in loop
float v = ... //input value
v = myLookupTable[static_cast<int>(v * LUT_RATIO)];

我正在尝试什么：

//outside loop
const __m128 LUT_RATIO = _mm_set1_ps(1000.0F);
//in loop
__m128 v = _mm_set_ps(v1, v2, v3, v4); //input values
__m128i vI = _mm_cvtps_epi32(_mm_mul_ps(v, LUT_RATIO)); //multiply and convert to integers
v = ??? // how to get vI indices of myLookupTable?

编辑：Ildjarn提出了一个需要我澄清的观点。我并不是要实现查找表代码的加速，我只是想避免将寄存器存储回浮点数专门用于查找，因为这部分夹在理论上可以从 SSE 中受益的其他 2 个部分之间。

如果你能等到明年，那么英特尔的Haswell CPU将拥有AVX2，其中包括收集负载的说明。这使您能够在一个指令中执行例如 8 个并行 LUT 查找（例如 VGATHERDPS）。除此之外，除非你的LUT非常小（例如16个元素），否则你就不走运了，在这种情况下，你可以使用PSHUFB。

相关文章：