include/symbol.h

Go to the documentation of this file.

#ifndef SYMBOL_H
#define SYMBOL_H

//== INCLUDES =================================================================

#include <string>
#include <vector>

#include <error.h>

//== NAMESPACES ===============================================================

namespace gpufilter {

//== IMPLEMENTATION ===========================================================

template< class T >
void copy_to_symbol( const std::string &name,
                     const T &value ) {
    size_t size_storage;
    cudaGetSymbolSize(&size_storage, name.c_str());
    cuda_error("Invalid symbol '"+name+"'");

    if( sizeof(T) > size_storage )
        throw std::runtime_error("'"+name+"'"+" storage overflow");

    cudaMemcpyToSymbol(name.c_str(), &value, sizeof(T), 0, cudaMemcpyHostToDevice);
    cuda_error("Error copying '"+name+"' buffer to device");
}

inline void copy_to_symbol( const std::string &name,
                            unsigned long value ) {
    copy_to_symbol(name, (unsigned int)value);
}

inline void copy_to_symbol( const std::string &name,
                            long value ) {
    copy_to_symbol(name, (int)value);
}

template< class T >
void copy_to_symbol( const std::string &name,
                     const std::string &size_name,
                     const std::vector<T> &items ) {
    size_t size_storage;
    cudaGetSymbolSize(&size_storage, name.c_str());
    cuda_error("Invalid symbol '"+name+"'");

    size_t size = items.size()*sizeof(T);

    if( size > size_storage )
        throw std::runtime_error("'"+name+"'"+" storage overflow");

    cudaMemcpyToSymbol(name.c_str(),&items[0], size, 0,
                       cudaMemcpyHostToDevice);
    cuda_error("Error copying '"+name+"' buffer to device");

    if( !size_name.empty() )
        copy_to_symbol(size_name.c_str(), items.size());
}

template< class T >
void copy_to_symbol( const std::string &name,
                     const std::vector<T> &items ) {
    copy_to_symbol(name, "", items);
}

//=============================================================================
} // namespace gpufilter
//=============================================================================
#endif // SYMBOL_H
//=============================================================================
//vi: ai sw=4 ts=4