From 56045667956a8e87742d0de1732807af5888c79e Mon Sep 17 00:00:00 2001 From: Andrew Kwangwoong Park Date: Mon, 11 Sep 2023 17:21:09 +0900 Subject: [PATCH] [GPU] Minor fix to get correct input layout for dump layer (#19686) Signed-off-by: Andrew Park --- src/plugins/intel_gpu/src/graph/network.cpp | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/plugins/intel_gpu/src/graph/network.cpp b/src/plugins/intel_gpu/src/graph/network.cpp index ebfca1cc7f1..f1a0c6b0773 100644 --- a/src/plugins/intel_gpu/src/graph/network.cpp +++ b/src/plugins/intel_gpu/src/graph/network.cpp @@ -1331,9 +1331,10 @@ void network::execute_impl(const std::vector& events) { "_" + get_iteration_prefix(curr_iter) + layer_name + "_src" + std::to_string(i); auto input_mem = get_primitive(inst->id())->dep_memory_ptr(i); + auto dep = inst->dependencies().at(i); + auto input_layout = dep.first->get_output_layout(dep.second); GPU_DEBUG_IF(debug_config->dump_layers_binary) { // Binary dump : raw - auto input_layout = inst->get_input_layout(i); auto filename = get_file_path_for_binary_dump(input_layout, name); mem_lock lock(input_mem, get_stream()); @@ -1342,7 +1343,7 @@ void network::execute_impl(const std::vector& events) { debug_str_for_bin_load += (filename + ","); } else { log_memory_to_file(input_mem, - inst->get_input_layout(i), + input_layout, get_stream(), name, debug_config->dump_layers_raw);