4

我正在使用 SIMD 来计算快速求幂结果。我将时间与非 simd 代码进行比较。取幂是使用平方和乘法算法实现的。

普通（非simd）版本的代码： 

b = 1;  
for (i=WPE-1; i>=0; --i){  
    ew = e[i];  
    for(j=0; j<BPW; ++j){  
        b = (b * b) % p;  
        if (ew & 0x80000000U)  b = (b * a) % p;  
        ew <<= 1;  
    }  
}

SIMD 版本： 

   B.data[0] = B.data[1] = B.data[2] = B.data[3] = 1U;  
   P.data[0] = P.data[1] = P.data[2] = P.data[3] = p;  
   for (i=WPE-1; i>=0; --i) {  
      EW.data[0] = e1[i]; EW.data[1] = e2[i]; EW.data[2] = e3[i]; EW.data[3] = e4[i];  
      for (j=0; j<BPW;++j){  
         B.v *= B.v; B.v -= (B.v / P.v) * P.v;  
         EWV.v = _mm_srli_epi32(EW.v,31);  
         M.data[0] = (EWV.data[0]) ? a1 : 1U;  
         M.data[1] = (EWV.data[1]) ? a2 : 1U; 
         M.data[2] = (EWV.data[2]) ? a3 : 1U; 
         M.data[3] = (EWV.data[3]) ? a4 : 1U;  
         B.v *= M.v; B.v -= (B.v / P.v) * P.v;  
         EW.v = _mm_slli_epi32(EW.v,1);  
      }  
   }

问题是虽然计算正确，但 simd 版本比非 simd 版本花费更多时间。

请帮我调试原因。也欢迎任何有关 SIMD 编码的建议。

谢谢和问候，阿努普。

4

2 回答 2

4

for 循环中的所有函数都应该是 SIMD 函数，而不仅仅是两个。为您的 2 个函数设置参数所花费的时间不如原始示例（很可能由编译器优化）

于 2010-10-25T07:43:57.180 回答
1

32 位 int 数据的 SIMD 循环通常如下所示：

for (i = 0; i < N; i += 4)
{
    // load input vector(s) with data at array index i..i+3
    __m128 va = _mm_load_si128(&A[i]);
    __m128 vb = _mm_load_si128(&B[i]);

    // process vectors using SIMD instructions (i.e. no scalar code)
    __m128 vc = _mm_add_epi32(va, vb);

    // store result vector(s) at array index i..i+3
    _mm_store_si128(&C[i], vc);
}

如果您发现需要在循环内的标量代码和 SIMD 代码之间移动，那么您可能不会从 SIMD 优化中获得任何收益。

SIMD 编程中的大部分技能都来自于找到使算法与给定 SIMD 架构提供的有限数量的受支持指令和数据类型一起工作的方法。您通常需要利用数据集的先验知识来获得最佳性能，例如，如果您确定您的 32 位整数值实际上具有适合 16 位的范围，那么这将使乘法成为您的算法更容易实现。

于 2010-10-25T10:49:06.720 回答