GpuRgbMemoryFactory.h
397 Bytes
#include<map>
#include "GpuRgbMemory.hpp"
#include "../FFNvDecoder/cuda_kernels.h"
#include <mutex>
class GpuRgbMemoryFactory{
public:
GpuRgbMemoryFactory();
// 获取内存
GpuRgbMemory* GpuRgbMemory* getMemory(int size, string id, string gpuid) ;(int size, string id, string gpuid) GpuRgbMemory* getMemory(int size, string id, string gpuid) ;
// 显存置为未使用状态
void releaseMemory(string id);
private:
map<string, GpuRgbMemory*> memoryMap;
std::mutex mtx;
};