首先要了解什么是SIMD,官方一点就是用向量化指令加速程序运行,通俗一点就是说**,用一个框架去写代码,这个代码的运算速度和空间使用都很棒**。
网上对于原理的介绍有很多,我这里主要偏实战。
想要执行SIMD,首先要配置好GCC的环境,MINGW即可,然后编写代码,使用带参数的命令执行即可(AVX框架)。
第一次使用的情况下,直接使用AVX框架(MINGW里面自带的)
代码
#include <iostream>
#include <immintrin.h>
void vectorAdd(float* a, float* b, float* result, int size) {
const int vectorSize = 8; // 使用 AVX 指令集,每次处理 8 个 float 元素
// 向量化循环
for (int i = 0; i < size; i += vectorSize) {
// 加载向量
__m256 vec_a = _mm256_loadu_ps(a + i);
__m256 vec_b = _mm256_loadu_ps(b + i);
// 向量加法
__m256 vec_result = _mm256_add_ps(vec_a, vec_b);
// 存储结果
_mm256_storeu_ps(result + i, vec_result);
}
}
int main() {
const int size = 16;
float a[size] = {1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f, 13.0f, 14.0f, 15.0f, 16.0f};
float b[size] = {1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f, 13.0f, 14.0f, 15.0f, 16.0f};
float result[size];
vectorAdd(a, b, result, size);
// 打印结果
for (int i = 0; i < size; ++i) {
std::cout << result[i] << " ";
}
std::cout << std::endl;
return 0;
}
文件命名为test.cpp
编译命令
g++ -o test test.cpp -mavx
执行命令
.\test
即可看到结果,到这里,恭喜你进入了SIMD的世界。