para_linear_transform.h

Go to the documentation of this file.

#ifndef __PARA_LINEAR_TRANSFORM_H__
#define __PARA_LINEAR_TRANSFORM_H__
#include "source_base/kernels/math_kernel_op.h"
#include "source_base/module_device/device.h"
#include "source_base/module_device/memory_op.h"
#include "source_base/parallel_device.h"
 
#include <vector>
#ifdef __MPI
#include "mpi.h"
#endif
namespace hsolver
{
 
template <typename T, typename Device>
class PLinearTransform
{
  public:
    using syncmem_dev_op = base_device::memory::synchronize_memory_op<T, Device, Device>;
    using resmem_dev_op = base_device::memory::resize_memory_op<T, Device>;
    using setmem_dev_op = base_device::memory::set_memory_op<T, Device>;
    using delmem_dev_op = base_device::memory::delete_memory_op<T, Device>;
    ~PLinearTransform();
    int nproc_col = 1;
    int rank_col = 0;
    int nrowA = 0;
    int ncolA = 0;
    int ncolB = 0;
    int LDA = 0;
    bool localU = false;
#ifdef __MPI
    MPI_Comm col_world = MPI_COMM_NULL;
    std::vector<int> colA_loc;
    std::vector<int> start_colA;
    std::vector<int> start_colB;
    int max_colA = 0;
    int ncolA_glo = 0;
    int max_colB = 0;
#endif
 
    void set_dimension(const int nrowA,
                       const int ncolA,
                       const int ncolB,
                       const int LDA,
#ifdef __MPI
                       MPI_Comm col_world,
#endif
                       const bool localU);
 
    void act(const T alpha, const T* A, const T* U_global, const T beta, T* B);
 
#ifdef __MPI
  private:
    std::vector<T> A_tmp_;      // temperory memory for A
    std::vector<T> isend_tmp_;  // temperory memory for isend
    T* U_tmp_ = nullptr;        // temperory memory for U
    T* B_tmp_ = nullptr;        // temperory memory for B
    T* A_tmp_device_ = nullptr; // temperory pointer
#endif
};
} // namespace hsolver
#endif