| 847 | } |
| 848 | |
| 849 | blas::BlasSupport* GpuExecutor::CreateBlas() { |
| 850 | PluginRegistry *registry = PluginRegistry::Instance(); |
| 851 | port::StatusOr<PluginRegistry::BlasFactory> status = |
| 852 | registry->GetFactory<PluginRegistry::BlasFactory>(cuda::kCudaPlatformId, |
| 853 | plugin_config_.blas()); |
| 854 | if (!status.ok()) { |
| 855 | LOG(ERROR) << "Unable to retrieve BLAS factory: " |
| 856 | << status.status().error_message(); |
| 857 | return nullptr; |
| 858 | } |
| 859 | |
| 860 | return status.ValueOrDie()(this); |
| 861 | } |
| 862 | |
| 863 | dnn::DnnSupport* GpuExecutor::CreateDnn() { |
| 864 | PluginRegistry *registry = PluginRegistry::Instance(); |