natas_hw
/
openvino
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[GPU] Add set_arguments() call in case of dynamic dependencies of static inst (#21664) (#21884) 

Co-authored-by: Sergey Shlyapnikov <sergey.shlyapnikov@intel.com>
This commit is contained in:
Roman Lyamin 2023-12-28 14:48:13 +04:00 committed by GitHub
parent 4e15cc0b7f
commit 7d948bf596
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/plugins/intel_gpu/src/graph/primitive_inst.cpp
View File

@ -494,7 +494,7 @@ event::ptr primitive_inst::realloc_if_needed() {
for (auto user : get_user_insts()) {
// Since fake alignment is applicable for input tensor as well, make sure we allocate enough memory
// to prevemt reading beyound the allocated memory bounds
if (user->get_node().is_type<fully_connected>()) {
if (user->get_node().is_type<fully_connected>() && user->is_dynamic()) {
user->update_shape();
user->update_shape_done_by_other = true;
@ -1112,8 +1112,14 @@ event::ptr primitive_inst::execute(const std::vector<event::ptr>& events) {
update_shape_done_by_other = false; // reset
OPENVINO_ASSERT(_impl != nullptr, "[GPU] Implementation is nullptr for ", primitive_id, " primitive");
// Dynamic insts may reallocate its' output buffer, so we need to update kernel's args respectively
bool has_dynamic_dependencies_insts = std::any_of(_deps.begin(), _deps.end(),
[](const std::pair<std::shared_ptr<primitive_inst>, int32_t>& dep) {
return dep.first->is_dynamic();
});
// Output buffer may be changed under the following conditions, so we need to set args to kernel on each iteration
if ((is_dynamic() && need_args_update) || has_mutable_input() || is_output()) {
if ((is_dynamic() && need_args_update) || has_mutable_input() || is_output() || (!is_dynamic() && has_dynamic_dependencies_insts)) {
set_arguments();
}
on_execute();