cuda.cuh

Go to the documentation of this file.

 
#pragma once
 
#include "../core/common.hpp"
 
namespace CellularAutomata
{
    namespace cuda
    {
        void allocateCUDA(void** arrayPtr, size_t bytes);
 
        void allocateHost(void** arrayPtr, size_t bytes);
 
        void freeCUDA(void* array);
 
        void freeHost(void* array);
 
        void copyCUDA(void* src, Device from, void* dst, Device to, size_t bytes);
 
        template <typename T, typename Activation>
        void epoch(
            T* input, T* kernel, T* output, Activation fn,
            const unsigned int h, const unsigned int w, const unsigned int s, const bool r);
    }
}
 
#include "cuda.inl"