| 28 | |
| 29 | namespace singa { |
| 30 | std::pair<size_t, size_t> CnMemPool::GetMemUsage() { |
| 31 | size_t free, total; |
| 32 | auto status = cnmemMemGetInfo(&free, &total, NULL); |
| 33 | CHECK_EQ(status, cnmemStatus_t::CNMEM_STATUS_SUCCESS) |
| 34 | << cnmemGetErrorString(status); |
| 35 | return std::make_pair(free, total); |
| 36 | } |
| 37 | std::pair<size_t, size_t> CnMemPool::GetMemUsage(int id) { |
| 38 | CHECK_EQ(cudaSetDevice(id), cudaError_t::cudaSuccess); |
| 39 | size_t free, total; |