ATC version 2.0, date: Aug7

git-svn-id: svn://svn.icms.temple.edu/lammps-ro/trunk@10561 f3b2605a-c512-4ea7-a41b-209d697bcdaa

lib/atc/Vector.cpp

@@ -0,0 +1,34 @@
+#include "DenseVector.h"
+
+namespace ATC_matrix {
+///////////////////////////////////////////////////////////////////////////////
+//* performs a matrix-vector multiply with optional transposes BLAS version
+void MultMv(const Matrix<double> &A, const Vector<double> &v, DenseVector<double> &c,
+            const bool At, double a, double b)
+{
+  static char t[2] = {'N','T'};
+  char *ta=t+At;
+  int sA[2] = {A.nRows(), A.nCols()};  // sizes of A
+  int sV[2] = {v.size(), 1};           // sizes of v
+  
+  GCK(A, v, sA[!At]!=sV[0], "MultAB<double>: matrix-vector multiply");
+  if (c.size() != sA[At])
+  {
+    c.resize(sA[At]); // set size of C to final size
+    c.zero();
+  }
+  // get pointers to the matrix sizes needed by BLAS
+  int *M = sA+At;  // # of rows in op[A]  (op[A] = A' if At='T' else A)
+  int *N = sV+1;   // # of cols in op[B]
+  int *K = sA+!At; // # of cols in op[A] or # of rows in op[B]
+
+  double *pa=A.ptr(), *pv=v.ptr(), *pc=c.ptr();
+
+#ifdef COL_STORAGE
+  dgemm_(ta, t, M, N, K, &a, pa, sA, pv, sV, &b, pc, M); 
+#else
+  dgemm_(t, ta, N, M, K, &a, pv, sV+1, pa, sA+1, &b, pc, N); 
+#endif
+}
+
+} // end namespace