| 169 | } |
| 170 | |
| 171 | void CudaGPU::CopyToFrom(void* dst, const void* src, size_t nBytes, |
| 172 | CopyDirection direction, Context* ctx) { |
| 173 | // cudaMemcpy(dst, src, nBytes, copyKind[direction]); |
| 174 | cudaMemcpyAsync(dst, src, nBytes, copyKind[direction], ctx_.stream); |
| 175 | } |
| 176 | |
| 177 | size_t CudaGPU::GetAllocatedMem() { |
| 178 | if (pool_ != nullptr) { |
nothing calls this directly
no outgoing calls
no test coverage detected