File: solve_gpu.h

package info (click to toggle)
r-cran-randomfieldsutils 1.2.5-1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 1,956 kB
  • sloc: ansic: 7,119; cpp: 6,437; fortran: 3,403; makefile: 7; sh: 1
file content (17 lines) | stat: -rw-r--r-- 435 bytes parent folder | download | duplicates (2) 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
 


#ifndef RFutils_gpusolve
#define RFutils_gpusolve 1


int cholGPU(bool copy, double *matrix, Uint size, double *B, Uint rhs_cols,
     double *LogDet, double *RESULT);
void mgpuSolve(double *matrix, Uint individuals, double *vector);
void gpu_relmat_custom(Uint*, double*, Uint, Uint);
void gpu_relmat_cublas(Uint*, double*, Uint, Uint);

// #define PADDIM 4L
//#define BLOCKS 1024
#define THREADS_PER_BLOCK 1024 //2048 / 32

#endif