| 719 | } |
| 720 | |
| 721 | fft::FftSupport* GpuExecutor::CreateFft() { |
| 722 | PluginRegistry* registry = PluginRegistry::Instance(); |
| 723 | port::StatusOr<PluginRegistry::FftFactory> status = |
| 724 | registry->GetFactory<PluginRegistry::FftFactory>(rocm::kROCmPlatformId, |
| 725 | plugin_config_.fft()); |
| 726 | if (!status.ok()) { |
| 727 | LOG(ERROR) << "Unable to retrieve FFT factory: " |
| 728 | << status.status().error_message(); |
| 729 | return nullptr; |
| 730 | } |
| 731 | |
| 732 | return status.ValueOrDie()(this); |
| 733 | } |
| 734 | |
| 735 | rng::RngSupport* GpuExecutor::CreateRng() { |
| 736 | PluginRegistry* registry = PluginRegistry::Instance(); |