| 12 | #ifdef WASMEDGE_PLUGIN_WASI_NN_BACKEND_OPENVINO |
| 13 | |
| 14 | static Expect<WASINN::ErrNo> |
| 15 | GetDeviceString(WASINN::Device TargetDevice, |
| 16 | std::string &DeviceString) noexcept { |
| 17 | switch (TargetDevice) { |
| 18 | case Device::AUTO: |
| 19 | case Device::CPU: |
| 20 | DeviceString = "CPU"; |
| 21 | break; |
| 22 | case Device::GPU: |
| 23 | DeviceString = "GPU"; |
| 24 | break; |
| 25 | default: |
| 26 | spdlog::error("[WASI-NN] Unsupported device type"sv); |
| 27 | return WASINN::ErrNo::InvalidArgument; |
| 28 | } |
| 29 | return WASINN::ErrNo::Success; |
| 30 | } |
| 31 | |
| 32 | Expect<WASINN::ErrNo> load(WASINN::WasiNNEnvironment &Env, |
| 33 | Span<const Span<uint8_t>> Builders, |