#ifndef PRIV_H_GUARD #define PRIV_H_GUARD #ifdef __cplusplus extern "C" { #endif #include "gpu.h" #include "glbopts.h" #include "linalg.h" #include "scs.h" struct SCS_LIN_SYS_WORK { /* reporting */ scs_int tot_cg_its; scs_float total_solve_time; /* ALL BELOW HOSTED ON THE GPU */ scs_float *p; /* cg iterate, n */ scs_float *r; /* cg residual, n */ scs_float *Gp; /* G * p, n */ scs_float *bg; /* b, n */ scs_float *tmp_m; /* m, used in mat_vec */ scs_float *z; /* preconditioned */ scs_float *M; /* preconditioner */ ScsGpuMatrix *Ag; /* A matrix on GPU */ ScsGpuMatrix *Agt; /* A trans matrix on GPU */ /* CUDA */ cublasHandle_t cublas_handle; cusparseHandle_t cusparse_handle; }; #ifdef __cplusplus } #endif #endif