1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
#ifndef PARDIAGONALMATRIX_H
#define PARDIAGONALMATRIX_H
#include "mpi.h"
#include "MPI_Wrappers.h"
#include "MatrixLibrary.h"
#include "DiagonalMatrix.h"
#include "DenseMatrix.h"
namespace ATC_matrix {
/**
* @class ParDiagonalMatrix
* @brief Parallelized version of DiagonalMatrix class.
*/
template <typename T>
class ParDiagonalMatrix : public DiagonalMatrix<T> {
public:
ParDiagonalMatrix(MPI_Comm comm, INDEX rows=0, bool z=0)
: DiagonalMatrix<T>(rows, z), _comm(comm) {}
ParDiagonalMatrix(MPI_Comm comm, const DiagonalMatrix<T>& c)
: DiagonalMatrix<T>(c), _comm(comm) {}
ParDiagonalMatrix(MPI_Comm comm, const Vector<T>& v)
: DiagonalMatrix<T>(v), _comm(comm) {}
void MultAB(const Matrix<T> &B, DenseMatrix<T> &C) const;
private:
MPI_Comm _comm;
};
template<>
void ParDiagonalMatrix<double>::MultAB(const Matrix<double> &B, DenseMatrix<double> &C) const;
} // end namespace
#endif
|
