diff --git a/src/imagination/pco/pco_index.c b/src/imagination/pco/pco_index.c
index eb26342b00c..d1425c78eb1 100644
--- a/src/imagination/pco/pco_index.c
+++ b/src/imagination/pco/pco_index.c
@@ -82,9 +82,15 @@ bool pco_index(pco_shader *shader, bool skip_ssa)
          }
       }
 
-      /* TODO: */
-      /* pco_foreach_if_in_func */
-      /* pco_foreach_loop_in_func */
+      func->next_if = 0;
+      pco_foreach_if_in_func (pif, func) {
+         pif->index = func->next_if++;
+      }
+
+      func->next_loop = 0;
+      pco_foreach_loop_in_func (loop, func) {
+         loop->index = func->next_loop++;
+      }
 
       if (!skip_ssa) {
          _mesa_hash_table_u64_destroy(func->vec_infos);