From 29608eb38ed02634080a017338f147da751a68d6 Mon Sep 17 00:00:00 2001 From: hyunback Date: Sun, 6 Oct 2024 11:19:43 +0900 Subject: [PATCH] Fix memory pool dependency issue. We should check is_runtime_skippable condition using memory pool. Signed-off-by: hyunback --- src/plugins/intel_gpu/src/graph/primitive_inst.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/plugins/intel_gpu/src/graph/primitive_inst.cpp b/src/plugins/intel_gpu/src/graph/primitive_inst.cpp index b0f7c4db565728..930e890886424d 100644 --- a/src/plugins/intel_gpu/src/graph/primitive_inst.cpp +++ b/src/plugins/intel_gpu/src/graph/primitive_inst.cpp @@ -2122,7 +2122,7 @@ memory::ptr primitive_inst::allocate_output(engine& _engine, GPU_DEBUG_LOG << "[" << _node.id() << ": constant]" << std::endl; return ov::intel_gpu::allocate_memory_evenif_zero_bytes(_engine, layout, alloc_type, reset); } - } else if (!_node.can_share_buffer() || _node.can_be_optimized() || _node.is_output()) { + } else if (!_node.can_share_buffer() || (impl_params.can_be_optimized() || !_node.is_runtime_skippable()) || _node.is_output()) { GPU_DEBUG_LOG << "[" << _node.id() << ": output]" << std::endl; return ov::intel_gpu::allocate_memory_evenif_zero_bytes(_engine, layout, alloc_type, reset); } else {