include_starpu/skepu/maparray.h

Go to the documentation of this file.

#ifndef MAP_ARRAY_H
#define MAP_ARRAY_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/task.h"
#include "skepu/vector.h"
#include "src/operator_macros.h"
#include "src/exec_plan.h"

namespace skepu
{

template <typename MapArrayFunc, typename T, skepu_container_type c_type=SKEPU_VECTOR> 
class MapArray: public Task
{
    
typedef struct
{
    MapArray<MapArrayFunc,T, c_type> *pMapArr;
    bool parts;
}ArgumentBuffer;

public:

    MapArray(MapArrayFunc* mapArrayFunc);
    
    MapArray(MapArrayFunc* mapArrayFunc, Vector<T> *in1, Vector<T> *in2, Vector<T> *out, int _parts=1);

    ~MapArray();
    
    Vector<T>& getOutput()
    {
        return *output;
    } 
    
    void *get_result()
    {
        return output;
    }
    

    void setInput1(Vector<T> *in)
    {
        input1 = in;
    }
    
    void setInput2(Vector<T> *in)
    {
        input2 = in;
    }
    
    void setOutput(Vector<T> *out)
    {
        output = out;
    } 
    
    void setPartitions(int _parts)
    {
        parts = _parts;
    } 

    
    void run_async();

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

private:
    Environment<int>* m_environment;
    MapArrayFunc* m_mapArrayFunc;
    ExecPlan m_execPlan;
    
#ifdef USE_STARPU_HISTORY_MODEL    
    starpu_perfmodel_t *maparr_perf_model;
#endif
    
    
    starpu_codelet *codelet;

public:
    void operator()(Vector<T>& input1, Vector<T>& input2, Vector<T>& output);
    
    void operator()(Vector<T>& input1, Vector<T>& input2, Vector<T>& output, int parts);

public:
    static void cpu_func(void *buffers[], void *arg);

#ifdef SKEPU_OPENMP
public:
    static void omp_func(void *buffers[], void *arg);
#endif

#ifdef SKEPU_CUDA
public:
    static void cuda_func(void *buffers[], void *arg);
#endif

#ifdef SKEPU_OPENCL
public:
    static void opencl_func(void *buffers[], void *arg);
    
public:
    struct starpu_opencl_program opencl_codelet;
    char kernelNameArray[1024];
    
    void replaceText(std::string& text, std::string find, std::string replace);
    void createOpenCLProgram();

#endif

private:
    std::string perfmodel_str;
    
    Vector<T> *input1;
    Vector<T> *input2;
    Vector<T> *output;
    int parts;
    
};


















template <typename MapArrayFunc, typename T> 
class MapArray<MapArrayFunc, T, MATRIX>: public Task
{

public:

    MapArray(MapArrayFunc* mapArrayFunc);
    
    MapArray(MapArrayFunc* mapArrayFunc, Vector<T> *in1, Matrix<T> *in2, Matrix<T> *out, int _parts=1);

    ~MapArray();

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

private:
    Environment<int>* m_environment;
    MapArrayFunc* m_mapArrayFunc;
    ExecPlan m_execPlan;
    
#ifdef USE_STARPU_HISTORY_MODEL    
    starpu_perfmodel_t *maparr_matr_perf_model;
#endif

    Matrix<T>& getOutput()
    {
        return *output;
    } 
    
    void *get_result()
    {
        return output;
    }
    

    void setInput1(Vector<T> *in)
    {
        input1 = in;
    }
    
    void setInput2(Matrix<T> *in)
    {
        input2 = in;
    }
    
    void setOutput(Matrix<T> *out)
    {
        output = out;
    } 
    
    void setPartitions(int _parts)
    {
        parts = _parts;
    } 
    
    void run_async();
    
    void setConstant1(T constant1) {m_mapArrayFunc->setConstant1(constant1);}
    
    void setConstant2(T constant1) {m_mapArrayFunc->setConstant2(constant1);}
    
    starpu_codelet *codelet;

public:
    void operator()(Vector<T>& input1, Matrix<T>& input2, Matrix<T>& output, int parts=1);

public:
    static void cpu_func(void *buffers[], void *arg);

#ifdef SKEPU_OPENMP
public:
    static void omp_func(void *buffers[], void *arg);
#endif

#ifdef SKEPU_CUDA
public:
    static void cuda_func(void *buffers[], void *arg);
#endif

#ifdef SKEPU_OPENCL
public:
    static void opencl_func(void *buffers[], void *arg);
    
public:
    struct starpu_opencl_program opencl_codelet;
    char kernelNameArray[1024];
    
    void replaceText(std::string& text, std::string find, std::string replace);
    void createOpenCLProgram();

#endif

private:
    std::string perfmodel_str;
    
    Vector<T> *input1;
    Matrix<T> *input2;
    Matrix<T> *output;
    int parts;
};



}

#include "src/maparray.inl"

#include "src/maparray_cpu.inl"

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

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

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

#endif