1// RUN: mlir-opt -allow-unregistered-dialect %s | FileCheck %s 2 3// CHECK-LABEL: func @custom_region_names 4func @custom_region_names() -> () { 5 "test.polyfor"() ( { 6 ^bb0(%arg0: index, %arg1: index, %arg2: index): 7 "foo"() : () -> () 8 }) { arg_names = ["i", "j", "k"] } : () -> () 9 // CHECK: test.polyfor 10 // CHECK-NEXT: ^bb{{.*}}(%i: index, %j: index, %k: index): 11 return 12} 13 14// CHECK-LABEL: func @weird_names 15// Make sure the asmprinter handles weird names correctly. 16func @weird_names() -> () { 17 "test.polyfor"() ( { 18 ^bb0(%arg0: i32, %arg1: i32, %arg2: index): 19 "foo"() : () -> i32 20 }) { arg_names = ["a .^x", "0"] } : () -> () 21 // CHECK: test.polyfor 22 // CHECK-NEXT: ^bb{{.*}}(%a_.5Ex: i32, %_0: i32, %arg0: index): 23 // CHECK-NEXT: %0 = "foo"() 24 return 25} 26 27