| 472 | // ---------------------------------------------------------------------- |
| 473 | |
| 474 | Result<std::shared_ptr<CudaHostBuffer>> AllocateCudaHostBuffer(int device_number, |
| 475 | const int64_t size) { |
| 476 | ARROW_ASSIGN_OR_RAISE(auto manager, CudaDeviceManager::Instance()); |
| 477 | return manager->AllocateHost(device_number, size); |
| 478 | } |
| 479 | |
| 480 | Result<uintptr_t> GetDeviceAddress(const uint8_t* cpu_data, |
| 481 | const std::shared_ptr<CudaContext>& ctx) { |