skip all user format check when dynamic shape in get_preferred_format() to avoid endless recursive call (#18096)

This commit is contained in:
Wilson Seok
2023-06-20 10:52:58 +09:00
committed by GitHub
parent 429af27b86
commit 3519050ef0

View File

@@ -1688,7 +1688,8 @@ format layout_optimizer::get_preferred_format(program_node& node) {
} else if (node.is_type<resample>()) {
// if the resample is in the last part of the network and there are no users using blocked format,
// it is better to reorder to bfyx before resample is done.
if (all_users_simple_format_until_output(node, node, 0, 10)) {
// Skip all user format check when node is dynamic. It could cause endless recursive call in get_preferred_foramt()
if (!node.is_dynamic() && all_users_simple_format_until_output(node, node, 0, 10)) {
const auto& dim = format::dimension(node.get_output_layout().format);
expected = format::get_default_format(dim, false, false);
} else {