[tools] Allow RegClass/Bank in update_givaluetracking_test_checks.py (#141727)

Pierre-vh · web-flow · commit 01848731d333 · 2025-05-28T10:29:18.000+02:00
The script previously assumed an underscore after the :
diff --git a/llvm/test/tools/UpdateTestChecks/update_givaluetracking_test_checks/Inputs/const.mir b/llvm/test/tools/UpdateTestChecks/update_givaluetracking_test_checks/Inputs/const.mir
@@ -20,7 +20,7 @@ body:             |
   ; CHECK-NEXT: %4:_ KnownBits:???????????????????????????????? SignBits:1
     %0:gpr32 = COPY $w0
     %1:_(s32) = COPY %0
-    %named:_(s64) = G_ZEXT %1
+    %named:gpr64(s64) = G_ZEXT %1
     %3:_(s32), %4:_(s32) = G_UNMERGE_VALUES %named(s64)
     $w0 = COPY %3(s32)
     $w1 = COPY %4(s32)
diff --git a/llvm/test/tools/UpdateTestChecks/update_givaluetracking_test_checks/Inputs/const.mir.expected b/llvm/test/tools/UpdateTestChecks/update_givaluetracking_test_checks/Inputs/const.mir.expected
@@ -18,12 +18,12 @@ body:             |
   ; Note: This comment should not be removed, the check lines below should be updated
   ; CHECK-LABEL: name: @Test2
   ; CHECK-NEXT: %1:_ KnownBits:???????????????????????????????? SignBits:1
-  ; CHECK-NEXT: %named:_ KnownBits:00000000000000000000000000000000???????????????????????????????? SignBits:32
+  ; CHECK-NEXT: %named:gpr64 KnownBits:00000000000000000000000000000000???????????????????????????????? SignBits:32
   ; CHECK-NEXT: %3:_ KnownBits:???????????????????????????????? SignBits:1
   ; CHECK-NEXT: %4:_ KnownBits:00000000000000000000000000000000 SignBits:32
     %0:gpr32 = COPY $w0
     %1:_(s32) = COPY %0
-    %named:_(s64) = G_ZEXT %1
+    %named:gpr64(s64) = G_ZEXT %1
     %3:_(s32), %4:_(s32) = G_UNMERGE_VALUES %named(s64)
     $w0 = COPY %3(s32)
     $w1 = COPY %4(s32)
diff --git a/llvm/utils/update_givaluetracking_test_checks.py b/llvm/utils/update_givaluetracking_test_checks.py
@@ -22,7 +22,7 @@
 
 VT_FUNCTION_RE = re.compile(
     r"\s*name:\s*@(?P<func>[A-Za-z0-9_-]+)"
-    r"(?P<body>(\s*%[0-9a-zA-Z_]+:_\s*KnownBits:[01?]+\sSignBits:[0-9]+$)+)",
+    r"(?P<body>(\s*%[0-9a-zA-Z_]+:[A-Za-z0-9_-]+\s*KnownBits:[01?]+\sSignBits:[0-9]+$)+)",
     flags=(re.X | re.M),
 )
 

Original file line number	Diff line number	Diff line change
`@@ -22,7 +22,7 @@`
`22`	`22`
`23`	`23`	`VT_FUNCTION_RE = re.compile(`
`24`	`24`	`r"\sname:\s@(?P<func>[A-Za-z0-9_-]+)"`
`25`		`- r"(?P<body>(\s%[0-9a-zA-Z_]+:_\sKnownBits:[01?]+\sSignBits:[0-9]+$)+)",`
	`25`	`+ r"(?P<body>(\s%[0-9a-zA-Z_]+:[A-Za-z0-9_-]+\sKnownBits:[01?]+\sSignBits:[0-9]+$)+)",`
`26`	`26`	`flags=(re.X \| re.M),`
`27`	`27`	`)`
`28`	`28`