scalapack_wrapper.hpp

Go to the documentation of this file.

#ifndef EXPLICIT_MAPMAKING_SCALAPACK_WRAPPER_HPP_
#define EXPLICIT_MAPMAKING_SCALAPACK_WRAPPER_HPP_

#include <vector>

namespace mapmaking {

inline bool i_am_in_grid(const std::vector<int>& desc) { return desc[1] != -1; }

int numroc_w(int m, int mb_a, int myroc, int rsrc_a, int n_proc);

int descinit_w(std::vector<int>* desc, int m, int n, int mb, int nb, int irsrc, int icsrc, int icontxt, int lld);

void pdgemr2d_w(int m, int n, std::vector<double>* a, int ia, int ja, std::vector<int>* desca,
                std::vector<double>* b, int ib, int jb, std::vector<int>* descb, int ictxt);

void pdgemm_w(char transa, char transb, int m, int n, int k, double alpha,
              std::vector<double>* a, int ia, int ja, std::vector<int>* desca,
              std::vector<double>* b, int ib, int jb, std::vector<int>* descb,
              double beta, std::vector<double>* c, int ic, int jc, std::vector<int>* descc);

void pdgemv_w(char transa, int m, int n, double alpha,
              std::vector<double>* a, int ia, int ja, std::vector<int>* desca,
              std::vector<double>* x, int ix, int jx, std::vector<int>* descx, int incx,
              double beta, std::vector<double>* y, int iy, int jy, std::vector<int>* descy, int incy);

int dsyev_w(char jobz, char uplo, int n, std::vector<double>* a, int lda,
            std::vector<double>* w, std::vector<double>* work, int lwork);

int dgemv_w(char trans, int m, int n, double alpha, std::vector<double>* a, int lda,
            std::vector<double>* x, int inc_x,
             double beta, std::vector<double>* y, int inc_y);

int dgemv_w(char trans, int m, int n, double alpha, std::vector<double>* a, int lda,
            std::vector<double>* x, int inc_x);

int pdsyevr_w(char jobz, char range, char uplo, int n,
              std::vector<double>* a, int ia, int ja, std::vector<int>* desca,
              double vl, double vu, int il, int iu, int* m, int* nz, std::vector<double>* w,
              std::vector<double>* z, int iz, int jz, std::vector<int>* descz,
              std::vector<double>* work, int lwork, std::vector<int>* iwork, int liwork);

int sym_inv_ev(char uplo, std::vector<double>* a, double cn_threshold);

// Assumes
// - squared proc grid
// - the size of the blocks is as balanced as possible
// - first element the (1, 1) element of proc with coordinates (0, 0) 
// - the size of the eigenvalues vector w being the side of the full matrix
int psym_inv_ev(char uplo, std::vector<double>* a, std::vector<int>* desca,
               double cn_threshold, std::vector<double>* w, std::vector<double>* u,
               std::pair<int, int> coor_proc_grid, int side_grid);
}

#endif  // EXPLICIT_MAPMAKING_SCALAPACK_WRAPPER_HPP_