[GPU] register read/assign as shape agnostic. Minor fix

vladimir-paramuzov · Nov 8, 2023 · 457504a · 457504a
1 parent da4ff63
commit 457504a
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/src/plugins/intel_gpu/src/graph/impls/cpu/assign.cpp b/src/plugins/intel_gpu/src/graph/impls/cpu/assign.cpp
@@ -73,7 +73,8 @@ struct assign_impl : public typed_primitive_impl<assign> {
 namespace detail {
 
 attach_assign_impl::attach_assign_impl() {
-    implementation_map<assign>::add(impl_types::cpu, assign_impl::create, {});
+    implementation_map<assign>::add(impl_types::cpu, shape_types::dynamic_shape, assign_impl::create, {});
+    implementation_map<assign>::add(impl_types::cpu, shape_types::static_shape, assign_impl::create, {});
 }
 
 }  // namespace detail

diff --git a/src/plugins/intel_gpu/src/graph/impls/cpu/read_value.cpp b/src/plugins/intel_gpu/src/graph/impls/cpu/read_value.cpp
@@ -75,7 +75,8 @@ struct read_value_impl : public typed_primitive_impl<read_value> {
 namespace detail {
 
 attach_read_value_impl::attach_read_value_impl() {
-    implementation_map<read_value>::add(impl_types::cpu, read_value_impl::create, {});
+    implementation_map<read_value>::add(impl_types::cpu, shape_types::dynamic_shape, read_value_impl::create, {});
+    implementation_map<read_value>::add(impl_types::cpu, shape_types::static_shape, read_value_impl::create, {});
 }
 
 }  // namespace detail

diff --git a/src/plugins/intel_gpu/src/plugin/variable_state.cpp b/src/plugins/intel_gpu/src/plugin/variable_state.cpp
@@ -64,7 +64,7 @@ void VariableState::set_state(const ov::SoPtr<ov::ITensor>& state) {
 }
 
 void VariableState::update_device_buffer() {
-    if (m_layout.is_dynamic())
+    if (m_layout.is_dynamic() || m_layout.bytes_count() == 0)
         return;
 
     if (actual_size < m_layout.bytes_count()) {