| 889 | } |
| 890 | |
| 891 | rng::RngSupport* GpuExecutor::CreateRng() { |
| 892 | PluginRegistry *registry = PluginRegistry::Instance(); |
| 893 | port::StatusOr<PluginRegistry::RngFactory> status = |
| 894 | registry->GetFactory<PluginRegistry::RngFactory>(cuda::kCudaPlatformId, |
| 895 | plugin_config_.rng()); |
| 896 | if (!status.ok()) { |
| 897 | LOG(ERROR) << "Unable to retrieve RNG factory: " |
| 898 | << status.status().error_message(); |
| 899 | return nullptr; |
| 900 | } |
| 901 | |
| 902 | return status.ValueOrDie()(this); |
| 903 | } |
| 904 | |
| 905 | // TODO(rspringer): Remove in b/18544742. |
| 906 | bool GpuExecutor::SupportsDnn() const { return true; } |