| 395 | } |
| 396 | |
| 397 | Result<std::shared_ptr<io::RandomAccessFile>> CudaMemoryManager::GetBufferReader( |
| 398 | std::shared_ptr<Buffer> buf) { |
| 399 | if (*buf->device() != *device_) { |
| 400 | return Status::Invalid( |
| 401 | "CudaMemoryManager::GetBufferReader called on foreign buffer " |
| 402 | "for device ", |
| 403 | buf->device()->ToString()); |
| 404 | } |
| 405 | return std::make_shared<CudaBufferReader>(buf); |
| 406 | } |
| 407 | |
| 408 | Result<std::shared_ptr<io::OutputStream>> CudaMemoryManager::GetBufferWriter( |
| 409 | std::shared_ptr<Buffer> buf) { |