don't have CHECK-LABEL consume the whole line

AntonLydike · Jul 7, 2024 · e338243 · e338243
1 parent ba52069
commit e338243
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/filecheck/compiler.py b/filecheck/compiler.py
@@ -78,10 +78,6 @@ def compile_uops(
             # we don't do numerical substitutions yet
             raise NotImplementedError("Numerical substitutions not supported!")
 
-    # make sure CHECK-LABEL consumes entire line
-    if check.name == "LABEL":
-        expr.append(r"[^\n]*")
-
     print(f"given: {check.arg}")
     print(f"compiled: {re.compile(''.join(expr))}")
     return re.compile("".join(expr)), captures
diff --git a/tests/filecheck/checks/check-label.test b/tests/filecheck/checks/check-label.test
@@ -25,3 +25,11 @@ region_b:
 // CHECK-NEXT: op a
 // CHECK-NEXT: op b
 // CHECK-NEXT: op c
+
+
+@func(%arg0: i32)
+op %arg0
+
+// CHECK-LABEL: @func
+// CHECK-SAME: ([[arg:%[A-Za-z0-9]+]]: i32)
+// CHECK: op [[arg]]
diff --git a/...lecheck/checks/check-not/check-not-1.test → tests/filecheck/checks/check-not-1.test b/...lecheck/checks/check-not/check-not-1.test → tests/filecheck/checks/check-not-1.test