aboutsummaryrefslogtreecommitdiffhomepage
path: root/bench/btl/libs
diff options
context:
space:
mode:
authorGravatar Gael Guennebaud <g.gael@free.fr>2010-06-23 22:10:49 +0200
committerGravatar Gael Guennebaud <g.gael@free.fr>2010-06-23 22:10:49 +0200
commit98fec45d3c2dd1febcff6a2fbeb243d71f77e1d1 (patch)
treecf2e1d6d969a0ff9d83f1ee60c6e365c9e026b5c /bench/btl/libs
parent546587c7d31178e23dc5ba5f9a3f3f44d8a19785 (diff)
btl: add a trmm action and update eigen interface
Diffstat (limited to 'bench/btl/libs')
-rw-r--r--bench/btl/libs/C_BLAS/C_BLAS_interface.hh13
-rw-r--r--bench/btl/libs/C_BLAS/main.cpp2
-rw-r--r--bench/btl/libs/eigen2/eigen2_interface.hh11
-rw-r--r--bench/btl/libs/eigen2/main_matmat.cpp1
4 files changed, 21 insertions, 6 deletions
diff --git a/bench/btl/libs/C_BLAS/C_BLAS_interface.hh b/bench/btl/libs/C_BLAS/C_BLAS_interface.hh
index f07df1033..aeb4e4380 100644
--- a/bench/btl/libs/C_BLAS/C_BLAS_interface.hh
+++ b/bench/btl/libs/C_BLAS/C_BLAS_interface.hh
@@ -134,6 +134,7 @@ static char trans = 'T';
static char nonunit = 'N';
static char lower = 'L';
static char right = 'R';
+static char left = 'L';
static int intone = 1;
template<>
@@ -186,7 +187,7 @@ public :
cblas_srot(N,A,1,B,1,c,s);
#endif
}
-
+
static inline void atv_product(gene_matrix & A, gene_vector & B, gene_vector & X, int N){
#ifdef PUREBLAS
sgemv_(&trans,&N,&N,&fone,A,&N,B,&intone,&fzero,X,&intone);
@@ -274,7 +275,7 @@ public :
sgetc2_(&N, C, &N, ipiv, jpiv, &info);
}
-
+
static inline void hessenberg(const gene_matrix & X, gene_matrix & C, int N){
#ifdef PUREBLAS
@@ -338,6 +339,14 @@ public :
#endif
}
+ static inline void trmm(gene_matrix & A, gene_matrix & B, gene_matrix & X, int N){
+ #ifdef PUREBLAS
+ strmm_(&left, &lower, &notrans,&nonunit, &N,&N,&fone,A,&N,B,&N);
+ #else
+ cblas_strmm(CblasColMajor, CblasLeft, CblasLower, CblasNoTrans,CblasNonUnit, N,N,1,A,N,B,N);
+ #endif
+ }
+
};
diff --git a/bench/btl/libs/C_BLAS/main.cpp b/bench/btl/libs/C_BLAS/main.cpp
index fdffdc979..99f512f9b 100644
--- a/bench/btl/libs/C_BLAS/main.cpp
+++ b/bench/btl/libs/C_BLAS/main.cpp
@@ -54,6 +54,8 @@ int main()
bench<Action_trisolve<C_BLAS_interface<REAL_TYPE> > >(MIN_MM,MAX_MM,NB_POINT);
bench<Action_trisolve_matrix<C_BLAS_interface<REAL_TYPE> > >(MIN_MM,MAX_MM,NB_POINT);
+ bench<Action_trmm<C_BLAS_interface<REAL_TYPE> > >(MIN_MM,MAX_MM,NB_POINT);
+
bench<Action_cholesky<C_BLAS_interface<REAL_TYPE> > >(MIN_MM,MAX_MM,NB_POINT);
bench<Action_partial_lu<C_BLAS_interface<REAL_TYPE> > >(MIN_MM,MAX_MM,NB_POINT);
diff --git a/bench/btl/libs/eigen2/eigen2_interface.hh b/bench/btl/libs/eigen2/eigen2_interface.hh
index d8d1a607f..e9d15ce8d 100644
--- a/bench/btl/libs/eigen2/eigen2_interface.hh
+++ b/bench/btl/libs/eigen2/eigen2_interface.hh
@@ -197,6 +197,10 @@ public :
X = L.template triangularView<Lower>().solve(B);
}
+ static inline void trmm(const gene_matrix & L, const gene_matrix& B, gene_matrix& X, int N){
+ X = L.template triangularView<Lower>() * B;
+ }
+
static inline void cholesky(const gene_matrix & X, gene_matrix & C, int N){
C = X;
ei_llt_inplace<Lower>::blocked(C);
@@ -211,8 +215,8 @@ public :
}
static inline void partial_lu_decomp(const gene_matrix & X, gene_matrix & C, int N){
- RowVectorXi piv(N);
- int nb;
+ Matrix<DenseIndex,1,Dynamic> piv(N);
+ DenseIndex nb;
C = X;
ei_partial_lu_inplace(C,piv,nb);
// C = X.partialPivLu().matrixLU();
@@ -221,8 +225,7 @@ public :
static inline void tridiagonalization(const gene_matrix & X, gene_matrix & C, int N){
typename Tridiagonalization<gene_matrix>::CoeffVectorType aux(N-1);
C = X;
- Tridiagonalization<gene_matrix>::_compute(C, aux);
-// C = Tridiagonalization<gene_matrix>(X).packedMatrix();
+ ei_tridiagonalization_inplace(C, aux);
}
static inline void hessenberg(const gene_matrix & X, gene_matrix & C, int N){
diff --git a/bench/btl/libs/eigen2/main_matmat.cpp b/bench/btl/libs/eigen2/main_matmat.cpp
index bdebc0832..aae261c29 100644
--- a/bench/btl/libs/eigen2/main_matmat.cpp
+++ b/bench/btl/libs/eigen2/main_matmat.cpp
@@ -27,6 +27,7 @@ int main()
bench<Action_matrix_matrix_product<eigen2_interface<REAL_TYPE> > >(MIN_MM,MAX_MM,NB_POINT);
bench<Action_ata_product<eigen2_interface<REAL_TYPE> > >(MIN_MM,MAX_MM,NB_POINT);
bench<Action_aat_product<eigen2_interface<REAL_TYPE> > >(MIN_MM,MAX_MM,NB_POINT);
+ bench<Action_trmm<eigen2_interface<REAL_TYPE> > >(MIN_MM,MAX_MM,NB_POINT);
return 0;
}