ScaLapackMatrix.hpp

Go to the documentation of this file.

#ifndef SCA_LAPACK_MATRIX_DEFINED
#define SCA_LAPACK_MATRIX_DEFINED
 
#include <util/BlacsWorld.hpp>
#include <util/Tensor.hpp>
 
namespace sisi4s {
class ScaLapackDescriptor {
public:
  ScaLapackDescriptor(BlacsWorld *blacsWorld, int lens_[2], int blockSize);
  int dataType; // 1 for dense matrices
  int blacsContext;
  int lens[2];
  int blockSize[2];
  int offset[2];
  int localLens[2]; // NOTE: assumes column major format
  // NOTE: the last element localLens[1]=number of local columns
  // is actually not part of the scalapack descriptor
};
 
template <typename F = double>
class ScaLapackMatrix : public ScaLapackDescriptor {
public:
  ScaLapackMatrix(ScaLapackMatrix<F> &A);
  ScaLapackMatrix(CTF::Matrix<F> &A,
                  BlacsWorld *blacsWorld,
                  int blockSize = 64);
  ScaLapackMatrix(Tensor<F> &A,
                  int lens[2],
                  BlacsWorld *blacsWorld,
                  int blockSize = 64);
  ~ScaLapackMatrix();
 
  const int *getDescriptor() const { return &dataType; }
 
  const F *getLocalValues() const { return localValues; }
 
  F *getLocalValues() { return localValues; }
 
  void write(CTF::Matrix<F> &A);
  void write(Tensor<F> &A);
 
protected:
  int getGlobalIndex(int localIndex, int dimension);
 
  BlacsWorld *blacsWorld;
  int64_t *localIndices;
 
  F *localValues;
};
} // namespace sisi4s
 
#endif