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

朱雀 2022-10-23 05:57 150阅读 0赞

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

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

1.  \#define WIN
2.  \#include "timing.h"
3.  \#include <intrin.h>
4.  \#include <stdlib.h>
5.  \#include <math.h>
6.  
7.  \#define N 4\*100000 // 注意：必须是4的倍数，否则使用SSE指令计算，要进行一些处理，从而保证正确。
8.  \_MM\_ALIGN16 **float** op1\[N\]; 
9.  \_MM\_ALIGN16 **float** op2\[N\]; 
10. \_MM\_ALIGN16 **float** result1\[N\]; 
11. \_MM\_ALIGN16 **float** result2\[N\]; 
12. 
13. **void** init() 
14. \{ 
15. **for**(**int** i = 0;i < N; i++) 
16.  \{ 
17.  op1\[i\] = (**float**)rand()/(**float**)RAND\_MAX; 
18.  op2\[i\] = (**float**)rand()/(**float**)RAND\_MAX; 
19.  \} 
20. \} 
21. 
22. **void** checkResult(**int** debug) 
23. \{ 
24. **bool** isSame = **true**; 
25. **for**(**int** i = 0;i < N; i++) 
26.  \{ 
27. **if** (debug) 
28.  \{ 
29.  printf("%lf %lf\\n", result1\[i\], result2\[i\]); 
30.  \} 
31. **else**
32.  \{ 
33. **if** (fabs(result1\[i\] - result2\[i\]) > 0.000001) 
34.  \{ 
35.  isSame = **false**; 
36. **break**; 
37.  \} 
38.  \} 
39.  \} 
40. **if** (!debug) \{ 
41. **if** (isSame) 
42.  printf("Result is Same\\n"); 
43. **else**
44.  printf("Result is not same\\n"); 
45.  \} 
46. \} 
47. 
48. **void** add1() 
49. \{ 
50. **for**(**int** i = 0; i < N;i++) 
51.  result1\[i\] = op1\[i\] + op2\[i\]; 
52. \} 
53. 
54. **void** add2() 
55. \{ 
56.  \_\_m128 a; 
57.  \_\_m128 b; 
58.  \_\_m128 c; 
59. 
60. **for**(**int** i = 0; i < N;i = i + 4) 
61.  \{ 
62. // Load
63.  a = \_mm\_load\_ps(op1 + i); 
64.  b = \_mm\_load\_ps(op2 + i); 
65. 
66.  c = \_mm\_add\_ps(a, b); // c = a + b
67. 
68.  \_mm\_store\_ps(result2 + i, c); 
69.  \} 
70. \} 
71. 
72. **int** main(**int** argc, **char**\* argv\[\]) 
73. \{ 
74.  init(); 
75.  srand((unsigned **int**)time(NULL)); 
76. 
77.  printf("Add a vector:\\n"); 
78.  startTiming(); 
79.  add1(); 
80.  stopWithPrintTiming(); 
81. 
82.  printf("\\n"); 
83.  printf("Add a vector with SSE instructions:\\n"); 
84.  startTiming(); 
85.  add2(); 
86.  stopWithPrintTiming(); 
87. 
88.  printf("\\n"); 
89.  checkResult(0); 
90. 
91. **return** 0; 
92. \}

![0_13221936047OiF.gif][]

[http_blog.csdn.net_gengshenghong_article_details_6973086]: http://blog.csdn.net/gengshenghong/article/details/6973086
[0_13221936047OiF.gif]: /images/20221023/76bbc9155d154fd98a00fb2b8d91ec9a.png