在C/C++代码中使用SSE等指令集的指令(5)SSE进行加法运算简单的性能测试

下面是一个简单的测试SSE指令性能的程序，可以看到明显的性能提升。

（说明：程序中的timing.h使用的是http://blog.csdn.net/gengshenghong/article/details/6973086中介绍的时间间隔获取方法）

#define WIN
#include "timing.h"
#include <intrin.h>
#include <stdlib.h>
#include <math.h>
 
#define N 4*100000        // 注意：必须是4的倍数，否则使用SSE指令计算，要进行一些处理，从而保证正确。
_MM_ALIGN16 float op1[N];
_MM_ALIGN16 float op2[N];
_MM_ALIGN16 float result1[N];
_MM_ALIGN16 float result2[N];
 
void init()
{
    for(int i = 0;i < N; i++)
    {
        op1[i] = (float)rand()/(float)RAND_MAX;
        op2[i] = (float)rand()/(float)RAND_MAX;
    }
}
 
void checkResult(int debug)
{
    bool isSame = true;
    for(int i = 0;i < N; i++)
    {
        if (debug)
        {
            printf("%lf        %lf\n", result1[i], result2[i]);
        }
        else
        {
            if (fabs(result1[i] - result2[i]) > 0.000001)
            {
                isSame = false;
                break;
            }
        }
    }
    if (!debug) {
        if (isSame)
            printf("Result is Same\n");
        else
            printf("Result is not same\n");
    }
}
 
void add1()
{
    for(int i = 0; i < N;i++)
        result1[i] = op1[i] + op2[i];
}
 
void add2()
{
    __m128  a;
    __m128  b;
    __m128  c;
 
    for(int i = 0; i < N;i = i + 4)
    {
        // Load
        a = _mm_load_ps(op1 + i);
        b = _mm_load_ps(op2 + i);
 
        c = _mm_add_ps(a, b);    // c = a + b
 
        _mm_store_ps(result2 + i, c);
    }
}
 
int main(int argc, char* argv[])
{
    init();
    srand((unsigned int)time(NULL));
 
    printf("Add a vector:\n");
    startTiming();
    add1();
    stopWithPrintTiming();
 
    printf("\n");
    printf("Add a vector with SSE instructions:\n");
    startTiming();
    add2();
    stopWithPrintTiming();
 
    printf("\n");
    checkResult(0);
 
    return 0;
}