2dmapoverlap.h

Go to the documentation of this file.

#ifndef MAPOVERLAP_2D_H
#define MAPOVERLAP_2D_H

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

#include "environment.h"
#include "../matrix.h"
#include "operator_macros.h"
#include "exec_plan.h"

namespace skepu
{


template <typename MapOverlap2DFunc>
class MapOverlap2D
{

public:

   MapOverlap2D(MapOverlap2DFunc* mapOverlapFunc);

   ~MapOverlap2D();

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

private:
   Environment<int>* m_environment;
   MapOverlap2DFunc* m_mapOverlapFunc;

   ExecPlan *m_execPlanMulti;

   ExecPlan *m_execPlan;

   ExecPlan m_defPlan;

public:
   template <typename T>
   void operator()(Matrix<T>& input);

   template <typename T>
   void operator()(Matrix<T>& input, Matrix<T>& output);

   template <typename T>
   void operator()(Matrix<T>& input, Matrix<T>& output, Matrix<T>& filter, bool useTiling=false);

   template <typename T>
   void operator()(Matrix<T>& input, Matrix<T>& output, size_t filter_rows, size_t filter_cols, bool useTiling=false);

public:
   template <typename T>
   void CPU(Matrix<T>& input);

   template <typename T>
   void CPU(Matrix<T>& input, Matrix<T>& output);

   template <typename T>
   void CPU(Matrix<T>& input, Matrix<T>& output, Matrix<T>& filter);

   template <typename T>
   void CPU(Matrix<T>& input, Matrix<T>& output, size_t filter_rows, size_t filter_cols);

#ifdef SKEPU_OPENMP
public:
   template <typename T>
   void OMP(Matrix<T>& input);

   template <typename T>
   void OMP(Matrix<T>& input, Matrix<T>& output);

   template <typename T>
   void OMP(Matrix<T>& input, Matrix<T>& output, Matrix<T>& filter);

   template <typename T>
   void OMP(Matrix<T>& input, Matrix<T>& output, size_t filter_rows, size_t filter_cols);
#endif

#ifdef SKEPU_CUDA
public:
   template <typename T>
   void CU(Matrix<T>& input, int useNumGPU = 1);

   template <typename T>
   void CU(Matrix<T>& input, Matrix<T>& output, int useNumGPU = 1);

   template <typename T>
   void CU(Matrix<T>& input, Matrix<T>& output, Matrix<T>& filter, bool useTiling=false, int useNumGPU=1);

   template <typename T>
   void CU(Matrix<T>& input, Matrix<T>& output, size_t filter_rows, size_t filter_cols, bool useTiling=false, int useNumGPU=1);

private:
   unsigned int cudaDeviceID;
   
   template <typename T>
   void mapOverlapSingleThread_CU(Matrix<T>& input, Matrix<T>& output, unsigned int deviceID);

   template <typename T>
   void mapOverlapMultipleThread_CU(Matrix<T>& input, Matrix<T>& output, size_t numDevices);
#endif

#ifdef SKEPU_OPENCL
public:
   template <typename T>
   void CL(Matrix<T>& input, int useNumGPU = 1);

   template <typename T>
   void CL(Matrix<T>& input, Matrix<T>& output, int useNumGPU = 1);

   template <typename T>
   void CL(Matrix<T>& input, Matrix<T>& output, Matrix<T>& filter, int useNumGPU = 1);

   template <typename T>
   void CL(Matrix<T>& input, Matrix<T>& output, size_t filter_rows, size_t filter_cols, int useNumGPU = 1);

private:
   template <typename T>
   void mapOverlapSingleThread_CL(Matrix<T>& input, Matrix<T>& output, unsigned int deviceID);

   template <typename T>
   void mapOverlapMultipleThread_CL(Matrix<T>& input, Matrix<T>& output, size_t numDevices);

private:
   std::vector<std::pair<cl_kernel, Device_CL*> > m_kernels_2D_CL;

   std::vector<std::pair<cl_kernel, Device_CL*> > m_kernels_Mat_ConvolFilter_CL;
   std::vector<std::pair<cl_kernel, Device_CL*> > m_kernels_Mat_Convol_CL;

   void replaceText(std::string& text, std::string find, std::string replace);
   void createOpenCLProgram();
#endif

};





}

#include "2dmapoverlap.inl"

#include "2dmapoverlap_cpu.inl"

#ifdef SKEPU_OPENMP
#include "2dmapoverlap_omp.inl"
#endif

#ifdef SKEPU_OPENCL
#include "2dmapoverlap_cl.inl"
#endif

#ifdef SKEPU_CUDA
#include "2dmapoverlap_cu.inl"
#endif

#endif