diff --git a/modules/nvidia_plugin/src/memory_manager/model/cuda_memory_model_builder.cpp b/modules/nvidia_plugin/src/memory_manager/model/cuda_memory_model_builder.cpp index cfdcfc4d9..52a4cafe2 100644 --- a/modules/nvidia_plugin/src/memory_manager/model/cuda_memory_model_builder.cpp +++ b/modules/nvidia_plugin/src/memory_manager/model/cuda_memory_model_builder.cpp @@ -14,13 +14,13 @@ void MemoryModelBuilder::addAllocation(BufferID id, int producerIndex, int lastC auto res = offsets_.emplace(id, 0); OPENVINO_ASSERT(res.second, "ID is not unique!"); // Verify that "id" is unique. const int64_t aligned_size = static_cast(applyAllignment(bsize)); - boxes_.emplace_back(MemorySolver::Box{producerIndex, lastConsumerIndex, aligned_size, id}); + boxes_.emplace_back(ov::MemorySolver::Box{producerIndex, lastConsumerIndex, aligned_size, id}); } MemoryModel::Ptr MemoryModelBuilder::build() { - MemorySolver solver{boxes_}; + ov::MemorySolver solver{boxes_}; const size_t blob_size = solver.solve(); - for (auto& pair : offsets_) pair.second = solver.getOffset(pair.first); + for (auto& pair : offsets_) pair.second = solver.get_offset(pair.first); return std::make_shared(blob_size, offsets_); } diff --git a/modules/nvidia_plugin/src/memory_manager/model/cuda_memory_model_builder.hpp b/modules/nvidia_plugin/src/memory_manager/model/cuda_memory_model_builder.hpp index 829a70327..d0fa57ec4 100644 --- a/modules/nvidia_plugin/src/memory_manager/model/cuda_memory_model_builder.hpp +++ b/modules/nvidia_plugin/src/memory_manager/model/cuda_memory_model_builder.hpp @@ -5,7 +5,7 @@ #pragma once #include "memory_manager/model/cuda_memory_model.hpp" -#include "memory_solver.hpp" +#include "openvino/runtime/memory_solver.hpp" namespace ov { namespace nvidia_gpu { @@ -37,7 +37,7 @@ class MemoryModelBuilder { MemoryModel::Ptr build(); private: - std::vector boxes_; + std::vector boxes_; std::unordered_map offsets_; };