nnlib
GPU-accelerated, C/C++ neural network library.
Classes | Enumerations
cache.h File Reference
#include <cstdlib>
#include <stack>
#include <unordered_map>

Go to the source code of this file.

Classes

class  Cache
 

Enumerations

enum  DataLocation { HOST , DEVICE }
 Enumerate to specify where data is located. More...
 

Detailed Description

Author
Jan Warchocki
Date
23 April 2024

Enumeration Type Documentation

◆ DataLocation

Enumerate to specify where data is located.

Can be either HOST or DEVICE. In case it is set to HOST, the data is stored in RAM and is processed by the CPU. In case it is set to DEVICE, the data is in VRAM and processed by the GPU. The latter is only possible if CUDA is installed and there is a CUDA enabled GPU on the system.