Function DefaultCUDADeviceMapper

cpp/src/arrow/gpu/cuda_memory.cc:501–505

namespace {

Result<std::shared_ptr<MemoryManager>> DefaultCUDADeviceMapper(int64_t device_id) {
  ARROW_ASSIGN_OR_RAISE(auto device,
                        arrow::cuda::CudaDevice::Make(static_cast<int>(device_id)));
  return device->default_memory_manager();
}

bool RegisterCUDADeviceInternal() {
  DCHECK_OK(RegisterDeviceMapper(DeviceAllocationType::kCUDA, DefaultCUDADeviceMapper));