diff --git a/src/plugins/intel_npu/src/backend/include/zero_host_tensor.hpp b/src/plugins/intel_npu/src/backend/include/zero_host_tensor.hpp index 7150a428dd270a..65b438284dfd9b 100644 --- a/src/plugins/intel_npu/src/backend/include/zero_host_tensor.hpp +++ b/src/plugins/intel_npu/src/backend/include/zero_host_tensor.hpp @@ -33,7 +33,7 @@ class ZeroHostTensor : public ov::ITensor { std::shared_ptr get_impl() const; private: - std::shared_ptr m_impl; + std::shared_ptr _impl; }; } // namespace intel_npu diff --git a/src/plugins/intel_npu/src/backend/include/zero_remote_tensor.hpp b/src/plugins/intel_npu/src/backend/include/zero_remote_tensor.hpp index 6217e52c15dae0..60578f3de64ef0 100644 --- a/src/plugins/intel_npu/src/backend/include/zero_remote_tensor.hpp +++ b/src/plugins/intel_npu/src/backend/include/zero_remote_tensor.hpp @@ -25,6 +25,8 @@ class ZeroRemoteTensor final : public RemoteTensor { ov::intel_npu::MemType mem_type = ov::intel_npu::MemType::L0_INTERNAL_BUF, void* mem = nullptr); + void* get_original_memory() const; + ~ZeroRemoteTensor() override; private: diff --git a/src/plugins/intel_npu/src/backend/src/zero_host_tensor.cpp b/src/plugins/intel_npu/src/backend/src/zero_host_tensor.cpp index 94115bf296b049..3a92a239408a1f 100644 --- a/src/plugins/intel_npu/src/backend/src/zero_host_tensor.cpp +++ b/src/plugins/intel_npu/src/backend/src/zero_host_tensor.cpp @@ -13,41 +13,36 @@ ZeroHostTensor::ZeroHostTensor(const std::shared_ptr& contex const ov::element::Type element_type, const ov::Shape& shape, const Config& config) - : m_impl(std::make_shared(context, - init_structs, - element_type, - shape, - config, - ov::intel_npu::TensorType::BINDED, - ov::intel_npu::MemType::L0_INTERNAL_BUF)) {} + : _impl(std::make_shared(context, + init_structs, + element_type, + shape, + config, + ov::intel_npu::TensorType::BINDED, + ov::intel_npu::MemType::L0_INTERNAL_BUF)) {} void* ZeroHostTensor::data(const ov::element::Type&) const { - auto itrHandle = m_impl->get_properties().find(ov::intel_npu::mem_handle.name()); - if (itrHandle == m_impl->get_properties().end()) { - OPENVINO_THROW("No parameter ", ov::intel_npu::mem_handle.name(), " found in parameters map"); - } - - return ov::Any(itrHandle->second).as(); + return _impl->get_original_memory(); } const ov::element::Type& ZeroHostTensor::get_element_type() const { - return m_impl->get_element_type(); + return _impl->get_element_type(); } const ov::Shape& ZeroHostTensor::get_shape() const { - return m_impl->get_shape(); + return _impl->get_shape(); } const ov::Strides& ZeroHostTensor::get_strides() const { - return m_impl->get_strides(); + return _impl->get_strides(); } void ZeroHostTensor::set_shape(ov::Shape new_shape) { - m_impl->set_shape(new_shape); + _impl->set_shape(new_shape); } std::shared_ptr ZeroHostTensor::get_impl() const { - return m_impl; + return _impl; } } // namespace intel_npu diff --git a/src/plugins/intel_npu/src/backend/src/zero_remote_tensor.cpp b/src/plugins/intel_npu/src/backend/src/zero_remote_tensor.cpp index 6fa12b72567792..c218aa14dd10a1 100644 --- a/src/plugins/intel_npu/src/backend/src/zero_remote_tensor.cpp +++ b/src/plugins/intel_npu/src/backend/src/zero_remote_tensor.cpp @@ -168,4 +168,8 @@ void ZeroRemoteTensor::update_properties() { } } +void* ZeroRemoteTensor::get_original_memory() const { + return _data; +} + } // namespace intel_npu