get layout from get_output_layout in primitive_inst (#21294)

This commit is contained in:
Sungeun Kim 2023-11-27 18:02:02 +09:00 committed by GitHub
parent e52f922d35
commit 1a235ffe79
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -76,7 +76,7 @@ struct condition_impl : typed_primitive_impl<condition> {
if (instance.dependencies()[dep_idx].first->id() == input_external_node->first) {
output_events.push_back(events[dep_idx]);
output_mem_ptr = instance.input_memory_ptr(dep_idx);
output_layout = output_mem_ptr->get_layout();
output_layout = instance.dependencies()[dep_idx].first->get_output_layout();
break;
}
}