64 lines
1.5 KiB
C++
64 lines
1.5 KiB
C++
#ifndef H_OPENCL_GREENSFUNCTION
|
|
#define H_OPENCL_GREENSFUNCTION
|
|
|
|
#include <iostream>
|
|
#include <cmath>
|
|
|
|
#include "../Algorithms/GreensFunction.h"
|
|
#include "OpenCLBase.h"
|
|
|
|
class OpenCLGreensFunction : public GreensFunction {
|
|
|
|
private:
|
|
|
|
bool base_create;
|
|
OpenCLBase *m_base;
|
|
|
|
public:
|
|
|
|
/** Constructor with OpenCLBase argument */
|
|
OpenCLGreensFunction(OpenCLBase *base);
|
|
|
|
/** Default constructor */
|
|
OpenCLGreensFunction();
|
|
|
|
/** Destructor */
|
|
~OpenCLGreensFunction();
|
|
|
|
/** Load OpenCL kernel file containing greens function kernels.
|
|
* m_base takes the kernel file and compiles the OpenCL programm.
|
|
*/
|
|
int buildProgram();
|
|
|
|
/**
|
|
Info: calc itegral on device memory (taken from OPAL src code)
|
|
Return: success or error code
|
|
*/
|
|
int greensIntegral(void *tmpgreen, int I, int J, int K, int NI, int NJ,
|
|
double hr_m0, double hr_m1, double hr_m2,
|
|
int streamId = -1);
|
|
|
|
/**
|
|
Info: integration of rho2_m field (taken from OPAL src code)
|
|
Return: success or error code
|
|
*/
|
|
int integrationGreensFunction(void *rho2_m, void *tmpgreen, int I, int J, int K,
|
|
int streamId = -1);
|
|
|
|
/**
|
|
Info: mirror rho field (taken from OPAL src code)
|
|
Return: succes or error code
|
|
*/
|
|
int mirrorRhoField(void *rho2_m, int I, int J, int K, int streamId = -1);
|
|
|
|
/**
|
|
Info: multiply complex fields already on the GPU memory, result will be put in ptr1
|
|
Return: success or error code
|
|
*/
|
|
int multiplyCompelxFields(void *ptr1, void *ptr2, int size, int streamId = -1);
|
|
|
|
};
|
|
|
|
|
|
#endif H_OPENCL_GREENSFUNCTION
|