Unified_Buffer.cpp

#include "Unified_Buffer.h"
#include <utility>

template <typename T>
UnifiedBuffer<T>::UnifiedBuffer()
{
    A = 0;
    B = 0;
    nbytes = 0;
}

template <typename T>
UnifiedBuffer<T>::UnifiedBuffer(int nelements)
{
    nbytes = sizeof(T) * nelements;
    cudaMalloc((void**)&A, nbytes);
    cudaMalloc((void**)&B, nbytes);
    if (0 == A || 0 == B)
    {
        printf("couldn't allocate GPU memory\n");
        return;
    }
    printf("Allocated %.2f MB on GPU\n", 2 * nbytes / (1024.f * 1024.f));
}

template <typename T>
T* UnifiedBuffer<T>::readTarget()
{
    return A;
}

template <typename T>
T* UnifiedBuffer<T>::writeTarget()
{
    return B;
}

template <typename T>
void UnifiedBuffer<T>::swap()
{
    std::swap(A, B);
}

template <typename T>
int UnifiedBuffer<T>::byteCount()
{
    return nbytes;
}

template <typename T>
UnifiedBuffer<T>::~UnifiedBuffer()
{
    cudaFree(A);
    cudaFree(B);
}