MPI_jacsolve
jacsolve_gpu.h
Go to the documentation of this file.
1 #ifndef JACSOLVEGPU_FILE
2 #define JACSOLVEGPU_FILE
3 #include <mpi.h>
4 
26 void JacobiSolve_GPU_4(const int nnode, const int nx, const int ny,
27  const int neigh[], const int color, const MPI::Intracomm& icomm,
28  const int id_d[], const int ik_d[], const float sk_d[],
29  const float f_d[], float u_d[],
30  float r_d[], float w_d[], float dd_d[],
31  const dim3 dimGrid, const dim3 dimBlock );
44 void JacobiSolve_GPU(const int nnode, const int id[], const int ik[], const float sk[],
45  const float f[], float u[] );
46 
55 float dscapr_GPU(const int n, const float x[], const float y[]);
56 
57 
66 void vddiv_GPU(const int n, float x[], const float y[], const float z[]);
67 
68 
83 void vdaxpy_GPU(const int n, float x[], const float y[],
84  const float a, const float z[] );
85 
86 
87 void set_value_GPU(const int n, float x[], const float a);
88 
89 
90 void Defect_GPU(float w[], const float f[], const float u[],
91  const int nnode, const int id[], const int ik[], const float sk[]);
92 
93 
103 void GetDiag_GPU(const int nnode, const int id[], const int ik[], const float sk[], float d[]);
104 
105 
106 #endif
107 
void GetDiag_GPU(const int nnode, const int id[], const int ik[], const float sk[], float d[])
void vdaxpy_GPU(const int n, float x[], const float y[], const float a, const float z[])
Element-wise vector multiplication x = y*z. A stride iy =0 for the source stride adds a constant valu...
void set_value_GPU(const int n, float x[], const float a)
void JacobiSolve_GPU(const int nnode, const int id[], const int ik[], const float sk[], const float f[], float u[])
float dscapr_GPU(const int n, const float x[], const float y[])
Calculates the Euclidian inner product of two vectors on GPU.
void vddiv_GPU(const int n, float x[], const float y[], const float z[])
Element-wise vector divison x = y/z.
void Defect_GPU(float w[], const float f[], const float u[], const int nnode, const int id[], const int ik[], const float sk[])
void JacobiSolve_GPU_4(const int nnode, const int nx, const int ny, const int neigh[], const int color, const MPI::Intracomm &icomm, const int id_d[], const int ik_d[], const float sk_d[], const float f_d[], float u_d[], float r_d[], float w_d[], float dd_d[], const dim3 dimGrid, const dim3 dimBlock)