Class SingleEigParam

XM/include/Dense/eig.h:9–32

Your code goes here single matrix eig parameter

// Your code goes here
// single matrix eig parameter
class SingleEigParam {
    public:
        int gpu_id;
        cusolverEigMode_t jobz;
        cublasFillMode_t uplo;
        cusolverDnParams_t param;

        SingleEigParam(const int gpu_id = 0): gpu_id(gpu_id) {
            CHECK_CUDA( cudaSetDevice(this->gpu_id) );
            this->param = NULL;
            CHECK_CUSOLVER( cusolverDnCreateParams(&this->param) );
            // we need eigen vector
            this->jobz = CUSOLVER_EIG_MODE_VECTOR;
            // upper and lower is arbitrary
            this->uplo = CUBLAS_FILL_MODE_LOWER;
        }

        ~SingleEigParam() {
            if (this->param != NULL) {
                CHECK_CUSOLVER( cusolverDnDestroyParams(this->param) );
                this->param = NULL;
            }
        }
};

// get single matrix eig buffer sizes in cusolver
template <typename T>