From ee31dc1c1b60eecdffdf94fa9a6b4a30a9c07c36 Mon Sep 17 00:00:00 2001 From: zhanmyz Date: Thu, 5 Dec 2024 16:58:36 +0800 Subject: [PATCH] add get openvino available ops function --- ggml/src/ggml-openvino.cpp | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/ggml/src/ggml-openvino.cpp b/ggml/src/ggml-openvino.cpp index 34d692a8cf..c25a927c30 100644 --- a/ggml/src/ggml-openvino.cpp +++ b/ggml/src/ggml-openvino.cpp @@ -458,6 +458,17 @@ static ggml_backend_buffer_t ggml_backend_openvino_device_buffer_from_host_ptr(g return nullptr; } +std::set get_openvino_available_opsets() { + ov::Core core; + std::set unique_ops; + for (const auto& opset : ov::get_available_opsets()) { + for (const auto& op : opset.second().get_type_info_set()) { + unique_ops.insert(op.name).second; + } + } + return unique_ops; +} + static bool ggml_backend_openvino_device_supports_op(ggml_backend_dev_t dev, const ggml_tensor * op) { GGML_ASSERT(dev->reg != nullptr); // ggml_backend_openvino_device_context * dev_ctx = (ggml_backend_openvino_device_context *) dev->context;