实现Multiply并通过avx编程添加2个矩阵

时间:2013-12-03 10:52:05

标签: c++ visual-studio-2012 matrix avx

我想使用AVX在Visual C ++ 2012中实现乘法和添加2个矩阵。我在Visual Studio中启用AVX(Advanced Vector Extensions (/arch:AVX))。但是当我启用此属性时添加矩阵以及禁用它时,时间相同并且启用此属性不会影响运行程序的时间。 对于4 * 4矩阵进行100000次迭代。时钟时间为9.启用或禁用Advanced Vector Extensions (/arch:AVX)此时没有更改。  另一个问题是如何通过AVX实现乘法2矩阵?

void AVXadd(
          double* pArray1,                   // [in] first source array
          double* pArray2,                   // [in] second source array
          double* pResult,                   // [out] result array
          int nSize)                        // [in] size of all arrays
{
    int nLoop = (nSize*nSize)/ 4;
    //

    __m256d* pSrc1 = (__m256d*) pArray1;
    __m256d* pSrc2 = (__m256d*) pArray2;
    __m256d* pDest = (__m256d*) pResult;
    for ( int i = 0; i < nLoop; i++ )
    {   
        *pDest = _mm256_add_pd(*pSrc1,*pSrc2); //add input arrays        
        pSrc1++;
        pSrc2++;
        pDest++;
    }
}


//get datas from random matrix and test add  
void AVX(vector<vector<double>> randn,int size,int itt){
    double *A,*B,*C;
    int ARRAY_SIZE=size*size;
    //alligment arrays:
    A = (double*) _aligned_malloc(ARRAY_SIZE * sizeof(double), 32);
    B = (double*) _aligned_malloc(ARRAY_SIZE * sizeof(double), 32);
    C = (double*) _aligned_malloc(ARRAY_SIZE * sizeof(double), 32);
    //fill by random vector numbers
    for(int i=0;i<size;i++)
        for(int j=0;j<size;j++)
            A[i*size+j]=B[i*size+j]=randn[i][j];//matrix to array

    clock_t t1, t2;
    t1=clock();
    //add
    for(int i=0;i<itt;i++)
        AVXadd(A,B,C,size);
    t2=clock();

    vector<vector<double>> c(size,vector<double>(size));
    for(int i=0;i<size;i++)
        for(int j=0;j<size;j++)
            c[i][j]=C[i*size+j];//C is result
    cout<<"\t\t"<<t2-t1;

}

0 个答案:

没有答案