[utils] simple refactor in update_mca_test_checks.py to make intent more readable

1 files changed, 13 insertions, 14 deletions

diff --git a/llvm/utils/update_mca_test_checks.py b/llvm/utils/update_mca_test_checks.py
index 64541641a75..c1f89e006fa 100755
--- a/llvm/utils/update_mca_test_checks.py
+++ b/llvm/utils/update_mca_test_checks.py
@@ -492,21 +492,20 @@ def _write_output(test_path, input_lines, prefix_list, block_infos,  # noqa
   output_check_lines = []
   used_prefixes = set()
   for block_num in range(len(block_infos)):
-    for block_text in sorted(block_infos[block_num]):
-      if not block_text:
-        continue
+    if type(block_infos[block_num]) is list:
+      # The block is of the type output from _break_down_block().
+      used_prefixes |= _write_block(output_check_lines,
+                                    block_infos[block_num],
+                                    not_prefix_set,
+                                    common_prefix,
+                                    prefix_pad)
+    else:
+      # _break_down_block() was unable to do do anything so output the block
+      # as-is.
+      for block_text in sorted(block_infos[block_num]):
+        if not block_text:
+          continue
 
-      if type(block_infos[block_num]) is list:
-        # The block is of the type output from _break_down_block().
-        used_prefixes |= _write_block(output_check_lines,
-                                      block_infos[block_num],
-                                      not_prefix_set,
-                                      common_prefix,
-                                      prefix_pad)
-        break
-      elif block_infos[block_num][block_text]:
-        # _break_down_block() was unable to do do anything so output the block
-        # as-is.
         lines = block_text.split('\n')
         for prefix in block_infos[block_num][block_text]:
           used_prefixes |= _write_block(output_check_lines,