| 331 | } |
| 332 | |
| 333 | static std::unique_ptr<DeviceMapperRegistryImpl> CreateDeviceRegistry() { |
| 334 | auto registry = std::make_unique<DeviceMapperRegistryImpl>(); |
| 335 | |
| 336 | // Always register the CPU device |
| 337 | DCHECK_OK(registry->RegisterDevice(DeviceAllocationType::kCPU, DefaultCPUDeviceMapper)); |
| 338 | |
| 339 | return registry; |
| 340 | } |
| 341 | |
| 342 | DeviceMapperRegistryImpl* GetDeviceRegistry() { |
| 343 | static auto g_registry = CreateDeviceRegistry(); |
no test coverage detected