Fix switch elimination test patterns - verified locally

Signed-off-by: devalgupta404 <devalgupta4@gmail.com>

Author: devalgupta404

test/Transforms/Passes/switch-elimination.mlir

@@ -12,71 +12,72 @@
 module {
   p4hir.control @c(%arg0: !p4hir.ref<!b32i> {p4hir.dir = #p4hir<dir inout>, p4hir.param_name = "b"})() {
     // Verify action functions are created with hidden annotation.
-    // CHECK-DAG: p4hir.func action @[[CASE_0:_switch_[0-9]+_case_0]]() annotations {hidden} {
+    // CHECK-DAG: p4hir.func action @_switch_{{[0-9]+}}_case_0() annotations {hidden} {
     // CHECK-DAG:   p4hir.return
     // CHECK-DAG: }
-    // CHECK-DAG: p4hir.func action @[[CASE_1:_switch_[0-9]+_case_1]]() annotations {hidden} {
+    // CHECK-DAG: p4hir.func action @_switch_{{[0-9]+}}_case_1() annotations {hidden} {
     // CHECK-DAG:   p4hir.return
     // CHECK-DAG: }
-    // CHECK-DAG: p4hir.func action @[[DEFAULT:_switch_[0-9]+_default]]() annotations {hidden} {
+    // CHECK-DAG: p4hir.func action @_switch_{{[0-9]+}}_default() annotations {hidden} {
     // CHECK-DAG:   p4hir.return
     // CHECK-DAG: }
 
     // Verify table structure with key, actions, default, and entries.
-    // CHECK: p4hir.table @[[TABLE:_switch_[0-9]+_table]] annotations {hidden} {
+    // CHECK: p4hir.table @_switch_{{[0-9]+}}_table annotations {hidden} {
     // CHECK:   p4hir.table_key
     // CHECK:     p4hir.match_key #exact
     // CHECK:   p4hir.table_actions {
-    // CHECK:     p4hir.table_action @[[CASE_0]]() {
-    // CHECK:       p4hir.call @c::@[[CASE_0]] () : () -> ()
+    // CHECK:     p4hir.table_action @_switch_{{[0-9]+}}_case_0() {
+    // CHECK:       p4hir.call @c::@_switch_{{[0-9]+}}_case_0 () : () -> ()
     // CHECK:     }
-    // CHECK:     p4hir.table_action @[[CASE_1]]() {
-    // CHECK:       p4hir.call @c::@[[CASE_1]] () : () -> ()
+    // CHECK:     p4hir.table_action @_switch_{{[0-9]+}}_case_1() {
+    // CHECK:       p4hir.call @c::@_switch_{{[0-9]+}}_case_1 () : () -> ()
     // CHECK:     }
-    // CHECK:     p4hir.table_action @[[DEFAULT]]() {
-    // CHECK:       p4hir.call @c::@[[DEFAULT]] () : () -> ()
+    // CHECK:     p4hir.table_action @_switch_{{[0-9]+}}_default() {
+    // CHECK:       p4hir.call @c::@_switch_{{[0-9]+}}_default () : () -> ()
     // CHECK:     }
     // CHECK:   }
     // CHECK:   p4hir.table_default_action {
-    // CHECK:     p4hir.call @c::@[[DEFAULT]] () : () -> ()
+    // CHECK:     p4hir.call @c::@_switch_{{[0-9]+}}_default () : () -> ()
     // CHECK:   }
     // CHECK:   p4hir.table_entries const {
     // CHECK:     p4hir.table_entry #p4hir.aggregate<[#int16_b32i]> : tuple<!b32i> {
-    // CHECK:       p4hir.call @c::@[[CASE_0]] () : () -> ()
+    // CHECK:       p4hir.call @c::@_switch_{{[0-9]+}}_case_0 () : () -> ()
     // CHECK:     }
     // CHECK:     p4hir.table_entry #p4hir.aggregate<[#int32_b32i]> : tuple<!b32i> {
-    // CHECK:       p4hir.call @c::@[[CASE_0]] () : () -> ()
+    // CHECK:       p4hir.call @c::@_switch_{{[0-9]+}}_case_0 () : () -> ()
     // CHECK:     }
     // CHECK:     p4hir.table_entry #p4hir.aggregate<[#int2_b32i]> : tuple<!b32i> {
-    // CHECK:       p4hir.call @c::@[[CASE_1]] () : () -> ()
+    // CHECK:       p4hir.call @c::@_switch_{{[0-9]+}}_case_1 () : () -> ()
     // CHECK:     }
     // CHECK:   }
     // CHECK: }
 
     // Verify table is applied directly with the condition (no intermediate variable).
-    // CHECK: p4hir.control_apply
-    // CHECK:   p4hir.read
-    // CHECK:   p4hir.table_apply
-    // CHECK:   p4hir.struct_extract
-    // CHECK:   p4hir.switch
-    // CHECK:     p4hir.case(equal
-    // CHECK:       p4hir.const #int1_b32i
-    // CHECK:       p4hir.assign
+    // CHECK: p4hir.control_apply {
+    // CHECK:   %{{.*}} = p4hir.read %arg0 : <!b32i>
+    // CHECK-NEXT: %{{.*}} = p4hir.table_apply @c::@_switch_{{[0-9]+}}_table with key{{.*}}%{{.*}}
+    // CHECK:   %{{.*}} = p4hir.struct_extract %{{.*}}["action_run"]
+    // CHECK:   p4hir.switch (%{{.*}}{{.*}}) {
+    // CHECK:     p4hir.case(equal, [#{{[_a-z0-9]+}}_switch_{{[0-9]+}}_case_0]) {
+    // CHECK:       %{{.*}} = p4hir.const #int1_b32i
+    // CHECK:       p4hir.assign %{{.*}}, %arg0 : <!b32i>
     // CHECK:       p4hir.yield
     // CHECK:     }
-    // CHECK:     p4hir.case(equal
-    // CHECK:       p4hir.const #int2_b32i
-    // CHECK:       p4hir.assign
+    // CHECK:     p4hir.case(equal, [#{{[_a-z0-9]+}}_switch_{{[0-9]+}}_case_1]) {
+    // CHECK:       %{{.*}} = p4hir.const #int2_b32i
+    // CHECK:       p4hir.assign %{{.*}}, %arg0 : <!b32i>
     // CHECK:       p4hir.yield
     // CHECK:     }
-    // CHECK:     p4hir.case(default
-    // CHECK:       p4hir.const #int3_b32i
-    // CHECK:       p4hir.assign
+    // CHECK:     p4hir.case(default, [#{{[_a-z0-9]+}}_switch_{{[0-9]+}}_default]) {
+    // CHECK:       %{{.*}} = p4hir.const #int3_b32i
+    // CHECK:       p4hir.assign %{{.*}}, %arg0 : <!b32i>
     // CHECK:       p4hir.yield
     // CHECK:     }
     // CHECK:     p4hir.yield
     // CHECK:   }
     // CHECK: }
+    
     p4hir.control_apply {
       %val = p4hir.read %arg0 : <!b32i>
       p4hir.switch (%val : !b32i) {
@@ -151,9 +152,9 @@ module {
 
     // The switch on action_run should remain unchanged.
     // CHECK: p4hir.control_apply {
-    // CHECK:   %[[APPLY:.*]] = p4hir.table_apply @c_action_run::@t with key()
-    // CHECK:   %[[AR:.*]] = p4hir.struct_extract %[[APPLY]]["action_run"]
-    // CHECK:   p4hir.switch (%[[AR]] : !anon) {
+    // CHECK:   %{{.*}} = p4hir.table_apply @c_action_run::@t with key()
+    // CHECK:   %{{.*}} = p4hir.struct_extract %{{.*}}["action_run"]
+    // CHECK:   p4hir.switch (%{{.*}} : !anon) {
     // CHECK:     p4hir.case(equal, [#anon_a1]) {
     // CHECK:       p4hir.yield
     // CHECK:     }