fccbe006
Hu Chunming
初步完成从解码到算法处理,任务管理...
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
|
#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;
};
|