kernels.h

Go to the documentation of this file.

//##############################################################################
//##############################################################################

#ifndef KERNELS_H
#define KERNELS_H
#include<stdio.h>

__device__ double2 subtract(double2 a, double2 b);

__device__ double2 add(double2 a, double2 b);

__device__ double2 pow(double2 a, int b);

__device__ double2 mult(double2 a, double b);

__device__ double2 mult(double2 a, double2 b);

__global__ void make_cufftDoubleComplex(double *in, double2 *out);

__device__ unsigned int getGid3d3d();

__device__ unsigned int getBid3d3d();

__device__ unsigned int getTid3d3d();

__global__ void is_eq(bool *a, bool *b, bool *ans);

//##############################################################################
//##############################################################################

__device__ double complexMagnitude(double2 in);

__global__ void complexMultiply(double2 *in1, double2 *in2, double2 *out);

__host__ __device__ double2 complexMultiply(double2 in1, double2 in2);

__device__ double2 make_complex(double in, int evolution_type);

__global__ void complexAbsSum(double2 *in1, double2 *in2, double *out);

__global__ void complexMagnitude(double2 *in, double *out);

__device__ double complexMagnitudeSquared(double2 in);

__global__ void complexMagnitudeSquared(double2 *in, double *out);

__global__ void complexMagnitudeSquared(double2 *in, double2 *out);

__device__ double2 conjugate(double2 in);

__device__ double2 realCompMult(double scalar, double2 comp);

//##############################################################################
//##############################################################################

__global__ void cMult(double2* in1, double2* in2, double2* out);

__global__ void cMultPhi(double2* in1, double* in2, double2* out);

__global__ void cMultDensity(double2* in1, double2* in2, double2* out, double dt, double mass, int gstate, double gDenConst);

__global__ void cMultDensity_ast(EqnNode_gpu *eqn, double2* in, double2* out,
                                 double dx, double dy, double dz, double time,
                                 int e_num, double dt, double mass, int gstate,
                                 double gDenConst);


__global__ void pinVortex(double2* in1, double2* in2, double2* out);


__global__ void vecMult(double2 *in, double *factor, double2 *out);

__global__ void l2_norm(double *in1, double *in2, double *in3, double *out);

__global__ void l2_norm(double2 *in1, double2 *in2, double2 *in3, double *out);

__global__ void l2_norm(double *in1, double *in2, double *out);

__global__ void l2_norm(double2 *in1, double2 *in2, double *out);

__global__ void scalarDiv(double2* in, double factor, double2* out);

__global__ void scalarDiv(double* in, double factor, double* out);

__global__ void scalarMult(double2* in, double factor, double2* out);

__global__ void scalarPow(double2* in, double param, double2* out);

__global__ void vecConjugate(double2 *in, double2 *out);

__global__ void scalarDiv1D(double2*, double2*);

__global__ void scalarDiv2D(double2*, double2*);

__global__ void scalarDiv_wfcNorm(double2* in, double dr, double* pSum, double2* out);

//##############################################################################

__global__ void reduce(double2* in, double* out);

__global__ void thread_test(double* input, double* output);

__global__ void multipass(double2* input, double2* output, int pass);

__global__ void multipass(double* input, double* output);

//##############################################################################

__global__ void angularOp(double omega, double dt, double2* wfc, double* xpyypx, double2* out);

__global__ void ast_mult(double *array, double *array_out, EqnNode_gpu *eqn,
                         double dx, double dy, double dz, double time,
                         int element_num);
__global__ void ast_cmult(double2 *array, double2 *array_out, EqnNode_gpu *eqn,
                          double dx, double dy, double dz, double time,
                          int element_num);
__global__ void ast_op_mult(double2 *array, double2 *array_out,
                            EqnNode_gpu *eqn,
                            double dx, double dy, double dz, double time,
                            int element_num, int evolution_type, double dt);

__device__ double2 real_ast(double val, double dt);

__device__ double2 im_ast(double val, double dt);

__global__ void zeros(bool *in, bool *out);

__global__ void set_eq(double *in1, double *in2);

__global__ void energyCalc(double2 *wfc, double2 *op, double dt, double2 *energy, int gnd_state, int op_space, double sqrt_omegaz_mass, double gDenConst);

inline __device__ double2 braKetMult(double2 in1, double2 in2);


__global__ void pSum(double* in1, double* output, int pass);


#endif