如何改进基于sse的矩阵乘法

时间:2012-10-18 23:30:23

标签: c++ performance matrix sse simd

float** matrix::mult(float** matrix1){
  float** result=new float *[n];
  int i,j,k;
  for(i=0;i<n;i++){
    result[i]=new float [n];
  }
  vect v1;
  vect v2;
  vect v3;
  vect total;
  clock_t start, end;
  start = clock();
  float result_ij=0;
  for(i=0;i<n;i++){   
    for(j=0;j<n;j++){
      result_ij=0;
      total.v=_mm_set1_ps(0);
      for(k=0;k<n;k=k+4){
        v1.v=_mm_set_ps(user_matrix[k][j],user_matrix[k+1][j],user_matrix[k+2][j],user_matrix[k+3][j]);
        v2.v=_mm_set_ps(matrix1[i][k],matrix1[i][k+1],matrix1[i][k+2],matrix1[i][k+3]);
        v3.v=_mm_mul_ps(v1.v,v2.v);
        total.v=_mm_add_ps(total.v,v3.v);
      }
      result[i][j]=total.a[1]+total.a[0]+total.a[2]+total.a[3];
    }
  }
  end = clock();
  cout<<(double)(end-start)/CLOCKS_PER_SEC<<endl;
  return result;
}

此代码与标量代码的速度完全相同。我不明白为什么会这么慢,用g ++编译,vect类型是一个联合。

union vect {
__m128 v;    
float a[4];  
} ;

对于矩阵作为多维数组,将其加载到SSE寄存器的最快方法是什么?

1 个答案:

答案 0 :(得分:2)

我不确定你是否想自己实现这个,但英特尔提供了很多有趣的代码:

http://www.intel.com/design/Pentiumiii/sml/index.htm

它们具有用于乘以4x4,6x6矩阵的代码,找到它们的倒数和其他矩阵。对于有和没有SSE的两个版本,它们也显示了一些基准测试等