cuda_manager.hpp

/* cuda_manager.hpp */
/* $Id$ */
#ifndef __cuda_manager_hpp
#define __cuda_manager_hpp

#ifdef U_CUDA_ENABLED

#include <cuda.h>
#include <cuda_runtime_api.h>
#include <dcf/ipcmsg_man.hpp>
#include <dcf/manager.hpp>

namespace dcf
{
  class CUDAManager : public dcf::Manager
  {
  public:
       
    CUDAManager();
    
    virtual ~CUDAManager();
    
    virtual dcf::Manager* clone() const;
    
    virtual size_t getUnitsCount() const;
    
    virtual dcf::error_t unitAlloc(int unit_id);
    
    virtual dcf::error_t unitFree(int unit_id);
    
    virtual dcf::error_t unitsFree();
    
    virtual dcf::error_t isAllocatedUnit(int unit_id, bool &is_allocated) const;
    
    dcf::error_t isAllocatedUnit(int unit_id, bool &is_allocated, IPCMsgMan::Message &udata) const;
    
    dcf::error_t currAllocatedUnit(int &unit_id) const;
    
    virtual dcf::error_t unitPower(int unit_id, double &power) const;
    
    dcf::error_t unitProp(int dev_id, cudaDeviceProp &dev_prop) const;
    
    dcf::error_t currentUnitProp(cudaDeviceProp &dev_prop) const;
    
    bool isInitialized() const {return m_is_initialized;}
    
    const dcf::IPCMsgMan& getMsgMan() const {return m_msg_man;}
    
  protected:
    
    cudaError_t cuda_dev_alloc(int dev_id) const;
    
    cudaError_t cuda_dev_free() const;
    
    int m_curr_allocated_unit;
    
    bool m_is_initialized;
    
    dcf::IPCMsgMan m_msg_man;
    
  private:

  };

};

#endif /* U_CUDA_ENABLED */

#endif /* cuda_manager.hpp */