mlir/test/IR/pretty-region-args.mlir

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