| 309 | #endif |
| 310 | |
| 311 | void InitDevices() { |
| 312 | CheckMemoryMethod(); |
| 313 | PADDLE_ENFORCE_NE(memory_method_->init_devices, |
| 314 | nullptr, |
| 315 | common::errors::Unavailable( |
| 316 | "init_devices method in memory_method_ is not " |
| 317 | "initialized yet. You need init it first.")); |
| 318 | memory_method_->init_devices(); |
| 319 | } |
| 320 | |
| 321 | void EmplaceDeviceContexts( |
| 322 | std::map<Place, std::shared_future<std::unique_ptr<DeviceContext>>>* |
no outgoing calls