[GPU] Forward further dependency events of skipped exec

Lyamin-Roman · Oct 14, 2024 · 5ce8c6b · 5ce8c6b
1 parent 1b4faa6
commit 5ce8c6b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/plugins/intel_gpu/src/graph/impls/ocl/gemm.cpp b/src/plugins/intel_gpu/src/graph/impls/ocl/gemm.cpp
@@ -157,7 +157,9 @@ struct gemm_impl : multi_stage_primitive<gemm> {
         if (instance.get_input_layout(0).count() == 0 ||
             instance.get_input_layout(1).count() == 0) {
             stream& stream = instance.get_network().get_stream();
-            stream.enqueue_barrier();
+            for (const auto& ev : events) {
+                ev->wait();
+            }
             return instance.output_memory_ptr()->fill(stream);
         }
 

diff --git a/src/plugins/intel_gpu/src/graph/primitive_inst.cpp b/src/plugins/intel_gpu/src/graph/primitive_inst.cpp
@@ -1526,7 +1526,7 @@ event::ptr primitive_inst::execute(const std::vector<event::ptr>& events) {
         }
 
         if (can_skip_execution) {
-            auto ev = get_network().get_stream().create_user_event(true);
+            auto ev = get_network().get_stream().aggregate_events(events);
             update_shape_done_by_other = false; // reset
             return ev;
         }