cache.h

Go to the documentation of this file.

 
#ifndef NNLIB_CACHE_H
#define NNLIB_CACHE_H
 
#include <cstdlib>
#include <stack>
#include <unordered_map>
 
enum DataLocation { HOST, DEVICE };
 
 
class Cache {
private:
    std::unordered_map<size_t, std::stack<float*>> hostCache;
    std::unordered_map<size_t, std::stack<float*>> deviceCache;
 
    Cache();
 
public:
    static Cache& getInstance() {
        static Cache instance;
        return instance;
    }
 
    float* get(size_t size, DataLocation location);
 
    void put(size_t size, float* ptr, DataLocation location);
 
    ~Cache();
 
    Cache(Cache const&) = delete;
    void operator=(Cache const&) = delete;
};
 
 
#endif //NNLIB_CACHE_H