| 691 | } |
| 692 | |
| 693 | blas::BlasSupport* GpuExecutor::CreateBlas() { |
| 694 | PluginRegistry* registry = PluginRegistry::Instance(); |
| 695 | port::StatusOr<PluginRegistry::BlasFactory> status = |
| 696 | registry->GetFactory<PluginRegistry::BlasFactory>(rocm::kROCmPlatformId, |
| 697 | plugin_config_.blas()); |
| 698 | if (!status.ok()) { |
| 699 | LOG(ERROR) << "Unable to retrieve BLAS factory: " |
| 700 | << status.status().error_message(); |
| 701 | return nullptr; |
| 702 | } |
| 703 | |
| 704 | return status.ValueOrDie()(this); |
| 705 | } |
| 706 | |
| 707 | dnn::DnnSupport* GpuExecutor::CreateDnn() { |
| 708 | PluginRegistry* registry = PluginRegistry::Instance(); |