devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

add intitial support for the vectorization of complex<float>

Browse Source
This commit is contained in:
Gael Guennebaud
2010-07-05 16:18:09 +02:00
parent efb79600b9
commit e1eccfad3f
6 changed files with 195 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
bench/bench_gemm.cpp
View File

@@ -10,7 +10,7 @@ using namespace std;
using namespace Eigen;
#ifndef SCALAR
#define SCALAR float
#define SCALAR std::complex<float>
#endif
typedef SCALAR Scalar;
@@ -26,6 +26,8 @@ static float fone = 1;
static float fzero = 0;
static double done = 1;
static double szero = 0;
static std::complex<float> cfone = 1;
static std::complex<float> cfzero = 0;
static char notrans = 'N';
static char trans = 'T';
static char nonunit = 'N';
@@ -44,6 +46,17 @@ void blas_gemm(const MatrixXf& a, const MatrixXf& b, MatrixXf& c)
c.data(),&ldc);
}
void blas_gemm(const MatrixXcf& a, const MatrixXcf& b, MatrixXcf& c)
{
int M = c.rows(); int N = c.cols(); int K = a.cols();
int lda = a.rows(); int ldb = b.rows(); int ldc = c.rows();
cgemm_(&notrans,&notrans,&M,&N,&K,(float*)&cfone,
const_cast<float*>((const float*)a.data()),&lda,
const_cast<float*>((const float*)b.data()),&ldb,(float*)&cfone,
(float*)c.data(),&ldc);
}
void blas_gemm(const MatrixXd& a, const MatrixXd& b, MatrixXd& c)
{
int M = c.rows(); int N = c.cols(); int K = a.cols();
@@ -98,7 +111,7 @@ int main(int argc, char ** argv)
}
if(cache_size>0)
setCpuCacheSizes(cache_size,32*cache_size);
setCpuCacheSizes(cache_size,96*cache_size);
int m = s;
int n = s;