C++SIMD初试

测试SIMD运算效率

• 代码如下：

#include <iostream>
#include "PxPhysicsAPI.h"
#include <immintrin.h>
#include <stdio.h>

// 打印向量p
void _mm256_print_epi32(__m256i p)
{
    int* p1 = (int*)&p;
    std::cout << p1[0] << " " << p1[1] << " " << p1[2]
        << " " << p1[3] << " " << p1[4] << " " << p1[5]
        << " " << p1[6] << " " << p1[7] << " " << std::endl;
}

// 运算i次8维加法(每次加法执行8组数相加)-返回运算时间
double add1(int i)
{
    __m256i a = _mm256_set_epi32(7, 6, 5, 4, 3, 2, 1, 0);// 定义一个8维向量
    clock_t start_time = clock();
    while (i-- > 0)
    {
        _mm256_add_epi32(a, a);// 进行一次8维运算
    }
    clock_t end_time = clock();
    return static_cast<double>(end_time - start_time) / CLOCKS_PER_SEC * 1000;
}

// 运算i次1维加法(每次加法执行1组数相加)-返回运算时间
double add2(int i)
{
    int b = 0;
    clock_t start_time = clock();
    while (i-- > 0)
    {
        b + b;
    }
    clock_t end_time = clock();
    return static_cast<double>(end_time - start_time) / CLOCKS_PER_SEC * 1000;
}

// 测试SIMD加法和普通加法的效率(运行c组普通加法)
void Test(int c)
{
    double time1 = add1(c/8);   // 等价于运行c/8组8维加法
    double time2 = add2(c);     // 运行c组1维加法
    std::cout << "time1:" << time1 << std::endl;
    std::cout << "time2:" << time2 << std::endl;
}
int main()
{
    // 运行1亿次一维加法测试
    Test(10000000000);
    return 0;
}

• 运行结果：
  

参考资料

• SIMD入门
• Intel官方文档