include/skepu/src/environment.h

Go to the documentation of this file.

#ifndef DEVICE_H
#define DEVICE_H

#ifndef SKEPU_NUMGPU
#define SKEPU_NUMGPU 0
#endif

#include <vector>

#ifdef SKEPU_OPENCL
#include "device_cl.h"
#include "device_mem_pointer_cl.h"
#endif

#ifdef SKEPU_CUDA
#include "device_cu.h"
#include "device_mem_pointer_cu.h"
#endif

namespace skepu
{

template <typename T>
class Environment
{

public:
    static Environment* getInstance();

    #ifdef SKEPU_OPENCL
    void finishAll_CL();
    std::vector<Device_CL*> m_devices_CL;
    #endif

    #ifdef SKEPU_CUDA
    void finishAll_CU();
    std::vector<Device_CU*> m_devices_CU;
    #endif

    void finishAll();

private:
    Environment();
    ~Environment();

    void init();

    #ifdef SKEPU_OPENCL
    void init_CL();
    #endif

    #ifdef SKEPU_CUDA
    void init_CU();
    #endif



private:
    static Environment* instance;

};

}

#include "environment.inl"

#endif