From 9e9cf72973f705a2a4b6beda782c8dea501d3c9b Mon Sep 17 00:00:00 2001 From: Eddy Kim Date: Thu, 3 Aug 2023 03:24:58 +0900 Subject: [PATCH] updated to reset variable memory when shape is changed (#18941) --- src/plugins/intel_gpu/src/graph/network.cpp | 1 + 1 file changed, 1 insertion(+) diff --git a/src/plugins/intel_gpu/src/graph/network.cpp b/src/plugins/intel_gpu/src/graph/network.cpp index c3784afbfc6..29f41c5d7e5 100644 --- a/src/plugins/intel_gpu/src/graph/network.cpp +++ b/src/plugins/intel_gpu/src/graph/network.cpp @@ -1694,6 +1694,7 @@ void network::update_variable_memory(const std::string& variable_id, const cldnn it->second->set_memory(get_engine().reinterpret_buffer(*it->second->memory, layout)); else it->second->set_memory(get_engine().allocate_memory(layout, false)); + it->second->is_set = false; } for (auto primitive : _variable_state_primitives) { if (const auto& memory_state_primitive = std::dynamic_pointer_cast(primitive)) {