Skip to content

Commit 93a02b0

Browse files
yeonbokmryzhov
andauthored
[GPU] Fix crash on set_output_memory due to shape difference b/w actual shape & user tensor (#29796)
### Details: - #29795 to release branch - Allow different output mem size with actual node size when user tenso is set and the port is remote - Fix crash in the wwb vlm test for phi-3.5 vision model ### Tickets: - *ticket-id* Co-authored-by: Mikhail Ryzhov <mikhail.ryzhov@intel.com>
1 parent eb51b54 commit 93a02b0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/plugins/intel_gpu/src/plugin/sync_infer_request.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -960,7 +960,7 @@ std::vector<cldnn::event::ptr> SyncInferRequest::prepare_output(size_t output_id
960960
auto output_tensor = std::dynamic_pointer_cast<RemoteTensorImpl>(m_plugin_outputs.at(output_idx).ptr);
961961
auto output_memory = output_tensor->get_memory();
962962
GPU_DEBUG_TRACE_DETAIL << internal_name << " with index " << output_idx << " prepare output: " << output_memory->buffer_ptr() << std::endl;
963-
return network->set_output_memory(internal_name, output_memory, is_remote_tensor_impl);
963+
return network->set_output_memory(internal_name, output_memory, is_dynamic && (is_remote_tensor_impl || user_tensor));
964964
}
965965

966966
void SyncInferRequest::init_mappings() {

0 commit comments

Comments
 (0)