generate.h

Go to the documentation of this file.

#ifndef GENERATE_H
#define GENERATE_H


#include <starpu.h>
#ifdef SKEPU_OPENCL 
  #include <starpu_opencl.h>
#endif  


#ifdef SKEPU_OPENCL
#include <string>
#include <vector>
#ifdef USE_MAC_OPENCL
#include <OpenCL/opencl.h>
#else
#include <CL/cl.h>
#endif
#include "src/device_cl.h"
#endif

#include "src/environment.h"

#include "skepu/vector.h"
#include "skepu/matrix.h"

#include "src/operator_macros.h"
#include "src/exec_plan.h"

namespace skepu
{

template <typename GenerateFunc>
class Generate
{

/* inf is different each part so for ebery part it will be created and destroyed */
typedef struct
{
    Generate<GenerateFunc> *pGenerate;
    int rowOffset;
    int colOffset;
}ArgumentBuffer;

public:

    Generate(GenerateFunc* generateFunc);

    ~Generate();

    template <typename U>
    void setConstant(U constant1) {m_generateFunc->setConstant(constant1);}

    void finishAll() {m_environment->finishAll();}
    void setExecPlan(ExecPlan& plan) {m_execPlan = plan;}

public:
    Environment<int>* m_environment;
    GenerateFunc* m_generateFunc;
    ExecPlan m_execPlan;
    
    struct starpu_codelet *codelet;

#ifdef USE_STARPU_HISTORY_MODEL    
    starpu_perfmodel *generate_perf_model;
#endif

public:
    template <typename T>
    void operator()(unsigned int numElements, Vector<T>& output, int parts=1);
    
    template <typename T>
    void operator()(unsigned int numRows, unsigned int numCols, Matrix<T>& output, int parts=1);

public:
    template <typename T>
    static void cpu_func(void *buffers[], void *arg);
    
    template <typename T>
    static void cpu_func_mat(void *buffers[], void *arg);

#ifdef SKEPU_OPENMP
public:
    template <typename T>
    static void omp_func(void *buffers[], void *arg);
    
    template <typename T>
    static void omp_func_mat(void *buffers[], void *arg);
#endif

#ifdef SKEPU_CUDA
public:
    template <typename T>
    static void cuda_func(void *buffers[], void *arg);
    
    template <typename T>
    static void cuda_func_mat(void *buffers[], void *arg);
#endif

#ifdef SKEPU_OPENCL
public:
    template <typename T>
    static void opencl_func(void *buffers[], void *arg);
    
    template <typename T>
    static void opencl_func_mat(void *buffers[], void *arg);
public:
    struct starpu_opencl_program opencl_program;
    char kernelNameArray[1024];
    
    void replaceText(std::string& text, std::string find, std::string replace);
    void createOpenCLProgram();
#endif

    std::string perfmodel_str;

};

}

#include "src/generate.inl"

#include "src/generate_cpu.inl"

#ifdef SKEPU_OPENMP
#include "src/generate_omp.inl"
#endif

#ifdef SKEPU_OPENCL
#include "src/generate_cl.inl"
#endif

#ifdef SKEPU_CUDA
#include "src/generate_cu.inl"
#endif

#endif