From 4ab5f0240a2bd823a14d1d42d5bbab1352160593 Mon Sep 17 00:00:00 2001
From: Mel Henning <drawoc@darkrefraction.com>
Date: Sat, 25 Jan 2025 12:27:08 -0500
Subject: [PATCH] nir: Update num_blocks in sort_unstructured_blocks

Cc: mesa-stable
Reviewed-by: Faith Ekstrand <faith.ekstrand@collabora.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/33219>
---
 src/compiler/nir/nir_control_flow.c | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/compiler/nir/nir_control_flow.c b/src/compiler/nir/nir_control_flow.c
index 6edf5e941f4..a427ff9c56a 100644
--- a/src/compiler/nir/nir_control_flow.c
+++ b/src/compiler/nir/nir_control_flow.c
@@ -939,6 +939,7 @@ nir_sort_unstructured_blocks(nir_function_impl *impl)
       assert(blocks[i].index == 0);
       assert(blocks[i].block == NULL);
    }
+   impl->num_blocks = count;
 
    foreach_list_typed_safe(nir_cf_node, node, node, &dead_blocks)
       cleanup_cf_node(node, impl);