Blame view

tsl_aiplatform/ai_platform/GpuRgbMemoryFactory.h 397 Bytes
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; 
  };