| 52 | const int kNumCudaStream = 1; |
| 53 | |
| 54 | CudaGPU::CudaGPU(int id) : Device(id, kNumCudaStream) { |
| 55 | MemPoolConf conf; |
| 56 | conf.add_device(id); |
| 57 | pool_ = std::make_shared<CnMemPool>(conf); |
| 58 | Setup(); |
| 59 | } |
| 60 | |
| 61 | CudaGPU::CudaGPU(int id, std::shared_ptr<DeviceMemPool> pool) |
| 62 | : Device(id, kNumCudaStream) { |