dngvd_op.h

Go to the documentation of this file.

// TODO: This is a temperary location for these functions.
// And will be moved to a global module(module base) later.
#ifndef MODULE_HSOLVER_DNGVD_H
#define MODULE_HSOLVER_DNGVD_H
 
#include "source_base/macros.h"
#include "source_base/module_external/lapack_wrapper.h"
#include "source_base/parallel_reduce.h"
#include "source_base/module_device/types.h"
 
namespace hsolver
{
 
inline double get_real(const std::complex<double> &x) { return x.real(); }
 
inline float get_real(const std::complex<float> &x) { return x.real(); }
 
inline double get_real(const double &x) { return x; }
 
inline float get_real(const float &x) { return x; }
 
 
template <typename T, typename Device>
struct dngvd_op
{
    using Real = typename GetTypeReal<T>::type;
    void operator()(const Device* d, const int nstart, const int ldh, const T* A, const T* B, Real* W, T* V);
};
 
template <typename T, typename Device>
struct dngv_op
{
    using Real = typename GetTypeReal<T>::type;
    void operator()(const Device* d, const int nstart, const int ldh, const T* A, T* B, Real* W, T* V);
};
 
template <typename T, typename Device>
struct dngvx_op
{
    using Real = typename GetTypeReal<T>::type;
    void operator()(const Device* d, const int nstart, const int ldh, T* A, T* B, const int m, Real* W, T* V);
};
 
template <typename T, typename Device>
struct dnevx_op
{
    using Real = typename GetTypeReal<T>::type;
    void operator()(const Device* d, const int nstart, const int ldh, const T* A, const int m, Real* W, T* V);
};
 
#if __CUDA || __UT_USE_CUDA || __ROCM || __UT_USE_ROCM
 
void createGpuSolverHandle();
void destroyGpuSolverHandle();
 
#endif
 
} // namespace hsolver
 
#endif // !MODULE_HSOLVER_DNGVD_H