include/skepu/src/device_cu.h

Go to the documentation of this file.

#ifndef DEVICE_CU_H
#define DEVICE_CU_H

#ifdef SKEPU_CUDA

#include <iostream>
#include <cuda.h>

namespace skepu
{

class Device_CU
{

public:
    cudaStream_t stream;
private:
    int m_deviceID;
    cudaDeviceProp m_deviceProp;
    int m_maxThreads;
    int m_maxBlocks;

    void getDeviceProps(int device)
    {
        cudaError_t err;
        err = cudaGetDeviceProperties(&m_deviceProp, device);
        if (err != cudaSuccess) {std::cerr<<"getDeviceProps failed!\n";}
    }

public:

    Device_CU(int id)
    {
        m_deviceID = id;
        getDeviceProps(id);

        if(m_deviceProp.major == 1 && m_deviceProp.minor < 2)
        {
            m_maxThreads = 256;
        }
        else
        {
            m_maxThreads = m_deviceProp.maxThreadsPerBlock;
        }

        m_maxBlocks = m_deviceProp.maxGridSize[0];
//        std::cerr<<"m_maxThreads: "<<m_maxThreads<<",  m_maxBlocks: "<<m_maxBlocks<<"\n";
    }

    ~Device_CU() {};

    int getMaxBlockSize() const {return m_deviceProp.maxThreadsPerBlock;}

    int getNumComputeUnits() const {return m_deviceProp.multiProcessorCount;}

    size_t getGlobalMemSize() const {return m_deviceProp.totalGlobalMem;}

    size_t getSharedMemPerBlock() const {return m_deviceProp.sharedMemPerBlock;}

    int getMaxThreads() const {return m_maxThreads;}

    int getMaxBlocks() const {return m_maxBlocks;}

    int getDeviceID() const {return m_deviceID;}
};

}

#endif

#endif