/home/brandes/workspace/LAMA/src/lama/cuda/CUDAContextManager.hpp

Go to the documentation of this file.

#ifndef LAMA_CUDA_CONTEXT_MANAGER_HPP_
#define LAMA_CUDA_CONTEXT_MANAGER_HPP_

// for dll_import
#include <lama/config.hpp>

// base classes
#include <lama/ContextManager.hpp>

// others
#include <lama/Context.hpp>

#include <lama/cuda/CUDAContext.hpp>

// boost
#include <boost/weak_ptr.hpp>

#include <string>

namespace lama
{

class CUDAContext;

#define LAMA_MAX_CUDA_DEVICES 4

class LAMA_DLL_IMPORTEXPORT CUDAContextManager : public ContextManager
{
public:

    ContextPtr getContext( int deviceNr )
    {
        return getInstance( deviceNr );
    }

    ~CUDAContextManager();

    static ContextPtr getInstance( int deviceNr );

private:

    CUDAContextManager();

    static boost::weak_ptr<CUDAContext> mCUDAContext [LAMA_MAX_CUDA_DEVICES];

    static CUDAContextManager theInstance; 
};

}

#endif // LAMA_CUDA_CONTEXT_MANAGER_HPP_