scal multi RHS

spm.c

@@ -26,6 +26,7 @@
 #include "p_spm.h"
 
 #include <cblas.h>
+#include <lapacke.h>
 
 #if !defined(DOXYGEN_SHOULD_SKIP_THIS)
 
@@ -1184,7 +1185,7 @@ spmCheckAxb( pastix_int_t nrhs,
  *
  *******************************************************************************/
 void
-spmScalMatrix(const double alpha, pastix_spm_t* spm)
+spmScalMatrix(double alpha, pastix_spm_t* spm)
 {
     switch(spm->flttype)
     {
@@ -1225,7 +1226,7 @@ spmScalMatrix(const double alpha, pastix_spm_t* spm)
  *
  *******************************************************************************/
 void
-spmScalVector(const double alpha, pastix_spm_t* spm, void *x)
+spmScalVector(double alpha, pastix_spm_t* spm, void *x)
 {
     switch(spm->flttype)
     {
@@ -1246,6 +1247,57 @@ spmScalVector(const double alpha, pastix_spm_t* spm, void *x)
     }
 }
 
+/**
+ *******************************************************************************
+ *
+ * @brief Scale a dense matrix corresponding to a set of RHS (wrapper to LAPACKE_xlascl)
+ *
+ * A = alpha * A
+ *
+ *******************************************************************************
+ *
+ * @param[in] flt
+ *          Datatype.
+ *
+ * @param[in] m
+ *          Number of rows of the matrix A.
+ *
+ * @param[in] n
+ *          Number of columns of the matrix A.
+ *
+ * @param[in] alpha
+ *           The scaling parameter.
+ *
+ * @param[inout] A
+ *          The matrix of RHS to scal.
+ *
+ * @param[in] lda
+ *          Defines the leading dimension of A when multiple right hand sides
+ *          are available. lda >= m.
+ *
+ *******************************************************************************/
+void
+spmScalRHS(pastix_coeftype_t flt, double alpha, pastix_int_t m, pastix_int_t n, void *A, pastix_int_t lda)
+{
+    switch(flt)
+    {
+    case PastixPattern:
+        break;
+    case PastixFloat:
+        LAPACKE_slascl_work(LAPACK_COL_MAJOR, 'G', 0, 0, 1., alpha, m, n, A, lda);
+        break;
+    case PastixComplex32:
+        LAPACKE_clascl_work(LAPACK_COL_MAJOR, 'G', 0, 0, 1., alpha, m, n, A, lda);
+        break;
+    case PastixComplex64:
+        LAPACKE_zlascl_work(LAPACK_COL_MAJOR, 'G', 0, 0, 1., alpha, m, n, A, lda);
+        break;
+    case PastixDouble:
+    default:
+        LAPACKE_dlascl_work(LAPACK_COL_MAJOR, 'G', 0, 0, 1., alpha, m, n, A, lda);
+    }
+}
+
 /**
  * @}
  */

spm.h

@@ -92,8 +92,9 @@ void          spmGenFakeValues( pastix_spm_t *spm );
  */
 double        spmNorm( pastix_normtype_t ntype, const pastix_spm_t *spm );
 int           spmMatVec( pastix_trans_t trans, const void *alpha, const pastix_spm_t *spm, const void *x, const void *beta, void *y );
-void          spmScalMatrix( const double alpha, pastix_spm_t *spm );
-void          spmScalVector( const double alpha, pastix_spm_t *spm, void *x );
+void          spmScalMatrix( double alpha, pastix_spm_t *spm );
+void          spmScalVector( double alpha, pastix_spm_t *spm, void *x );
+void          spmScalRHS( pastix_coeftype_t flt, double alpha, pastix_int_t m, pastix_int_t n, void *A, pastix_int_t lda );
 
 /**
  * @}