| 495 | } |
| 496 | |
| 497 | bool RegisterCUDADeviceInternal() { |
| 498 | DCHECK_OK(RegisterDeviceMapper(DeviceAllocationType::kCUDA, DefaultCUDADeviceMapper)); |
| 499 | // TODO add the CUDA_HOST and CUDA_MANAGED allocation types when they are supported in |
| 500 | // the CudaDevice |
| 501 | return true; |
| 502 | } |
| 503 | |
| 504 | static auto cuda_registered = RegisterCUDADeviceInternal(); |
| 505 |
no test coverage detected