xref: /llvm-project/mlir/test/IR/wrapping_op.mlir (revision 7143e3335d92b9f2a987cfccb65461e186af7f4b)
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"
7func.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}
16