MCPcopy Create free account
hub / github.com/apache/arrow / RegisterCUDADeviceInternal

Function RegisterCUDADeviceInternal

cpp/src/arrow/gpu/cuda_memory.cc:507–512  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

505}
506
507bool RegisterCUDADeviceInternal() {
508 DCHECK_OK(RegisterDeviceMapper(DeviceAllocationType::kCUDA, DefaultCUDADeviceMapper));
509 // TODO add the CUDA_HOST and CUDA_MANAGED allocation types when they are supported in
510 // the CudaDevice
511 return true;
512}
513
514static auto cuda_registered = RegisterCUDADeviceInternal();
515

Callers 1

cuda_memory.ccFile · 0.85

Calls 1

RegisterDeviceMapperFunction · 0.85

Tested by

no test coverage detected