| 505 | } |
| 506 | |
| 507 | bool 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 | |
| 514 | static auto cuda_registered = RegisterCUDADeviceInternal(); |
| 515 |
no test coverage detected