mlir/test/IR/wrapping_op.mlir

   1 // RUN: mlir-opt -allow-unregistered-dialect %s | FileCheck %s
   2 // RUN: mlir-opt -allow-unregistered-dialect -mlir-print-op-generic -mlir-print-debuginfo -mlir-print-local-scope %s | FileCheck %s --check-prefix=CHECK-GENERIC
   3
   4 // CHECK-LABEL: func @wrapping_op
   5 // CHECK-GENERIC: "func.func"
   6 // CHECK-GENERIC-SAME: sym_name = "wrapping_op"
   7 func.func @wrapping_op(%arg0 : i32, %arg1 : f32) -> (i3, i2, i1) {
   8 // CHECK: %0:3 = test.wrapping_region wraps "some.op"(%arg1, %arg0) {test.attr = "attr"} : (f32, i32) -> (i1, i2, i3)
   9 // CHECK-GENERIC: "test.wrapping_region"() ({
  10 // CHECK-GENERIC:   %[[NESTED_RES:.*]]:3 = "some.op"(%arg1, %arg0) {test.attr = "attr"} : (f32, i32) -> (i1, i2, i3) loc("some_NameLoc")
  11 // CHECK-GENERIC:   "test.return"(%[[NESTED_RES]]#0, %[[NESTED_RES]]#1, %[[NESTED_RES]]#2) : (i1, i2, i3) -> () loc("some_NameLoc")
  12 // CHECK-GENERIC: }) : () -> (i1, i2, i3) loc("some_NameLoc")
  13   %res:3 = test.wrapping_region wraps "some.op"(%arg1, %arg0) { test.attr = "attr" } : (f32, i32) -> (i1, i2, i3) loc("some_NameLoc")
  14   return %res#2, %res#1, %res#0 : i3, i2, i1
  15 }