Skip to content

Commit

Permalink
[PyOV] Add npu properties
Browse files Browse the repository at this point in the history
  • Loading branch information
akuporos committed Jan 13, 2025
1 parent 58b02fb commit 19278fb
Showing 1 changed file with 7 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,11 @@
#include "pyopenvino/graph/any.hpp"
#include "pyopenvino/utils/utils.hpp"

#include "openvino/runtime/intel_cpu/properties.hpp"
#include "openvino/runtime/intel_gpu/properties.hpp"
#include "openvino/runtime/intel_npu/properties.hpp"
#include "openvino/runtime/auto/properties.hpp"

namespace py = pybind11;

void regmodule_properties(py::module m) {
Expand Down Expand Up @@ -323,7 +328,8 @@ void regmodule_properties(py::module m) {

// Submodule npu
py::module m_intel_npu =
m_properties.def_submodule("intel_npu", "openvino.properties.intel_npu submodule that simulates ov::intel_npu");
m_properties.def_submodule("intel_npu",
"openvino.properties.intel_npu submodule that simulates ov::intel_npu");

wrap_property_RO(m_intel_npu, ov::intel_npu::device_alloc_mem_size, "device_alloc_mem_size");
wrap_property_RO(m_intel_npu, ov::intel_npu::device_total_mem_size, "device_total_mem_size");
Expand Down

0 comments on commit 19278fb

Please sign in to comment.