67 lines · plain
1// RUN: mlir-opt -allow-unregistered-dialect %s -pass-pipeline="builtin.module(func.func(test-clone))" --split-input-file | FileCheck %s2 3module {4 func.func @fixpoint(%arg1 : i32) -> i32 {5 %r = "test.use"(%arg1) ({6 %r2 = "test.use2"(%arg1) ({7 "test.yield2"(%arg1) : (i32) -> ()8 }) : (i32) -> i329 "test.yield"(%r2) : (i32) -> ()10 }) : (i32) -> i3211 return %r : i3212 }13}14 15// CHECK: notifyOperationInserted: test.use16// CHECK-NEXT: notifyOperationInserted: test.use217// CHECK-NEXT: notifyOperationInserted: test.yield218// CHECK-NEXT: notifyOperationInserted: test.yield19// CHECK-NEXT: notifyOperationInserted: func.return20 21// CHECK-LABEL: func @fixpoint22// CHECK-SAME: (%[[arg0:.+]]: i32) -> i32 {23// CHECK-NEXT: %[[i0:.+]] = "test.use"(%[[arg0]]) ({24// CHECK-NEXT: %[[r2:.+]] = "test.use2"(%[[arg0]]) ({25// CHECK-NEXT: "test.yield2"(%[[arg0]]) : (i32) -> ()26// CHECK-NEXT: }) : (i32) -> i3227// CHECK-NEXT: "test.yield"(%[[r2]]) : (i32) -> ()28// CHECK-NEXT: }) : (i32) -> i3229// CHECK-NEXT: %[[i1:.+]] = "test.use"(%[[i0]]) ({30// CHECK-NEXT: %[[r2:.+]] = "test.use2"(%[[i0]]) ({31// CHECK-NEXT: "test.yield2"(%[[i0]]) : (i32) -> ()32// CHECK-NEXT: }) : (i32) -> i3233// CHECK-NEXT: "test.yield"(%[[r2]]) : (i32) -> ()34// CHECK-NEXT: }) : (i32) -> i3235// CHECK-NEXT: return %[[i1]] : i3236// CHECK-NEXT: }37 38// -----39 40func.func @clone_unregistered_with_attrs() {41 "unregistered.foo"() <{bar = 1 : i64, flag = true, name = "test", value = 3.14 : f32}> : () -> ()42 "unregistered.bar"() : () -> ()43 "unregistered.empty_dict"() <{}> : () -> ()44 "unregistered.complex"() <{45 array = [1, 2, 3],46 dict = {key1 = 42 : i32, key2 = "value"},47 nested = {inner = {deep = 100 : i64}}48 }> : () -> ()49 return50}51 52// CHECK: notifyOperationInserted: unregistered.foo53// CHECK-NEXT: notifyOperationInserted: unregistered.bar54// CHECK-NEXT: notifyOperationInserted: unregistered.empty_dict55// CHECK-NEXT: notifyOperationInserted: unregistered.complex56// CHECK-NEXT: notifyOperationInserted: func.return57 58// CHECK-LABEL: func @clone_unregistered_with_attrs() {59// CHECK-NEXT: "unregistered.foo"() <{bar = 1 : i64, flag = true, name = "test", value = [[PI:.+]] : f32}> : () -> ()60// CHECK-NEXT: "unregistered.bar"() : () -> ()61// CHECK-NEXT: "unregistered.empty_dict"() <{}> : () -> ()62// CHECK-NEXT: "unregistered.complex"() <{array = [1, 2, 3], dict = {key1 = 42 : i32, key2 = "value"}, nested = {inner = {deep = 100 : i64}}}> : () -> ()63// CHECK-NEXT: "unregistered.foo"() <{bar = 1 : i64, flag = true, name = "test", value = [[PI]] : f32}> : () -> ()64// CHECK-NEXT: "unregistered.bar"() : () -> ()65// CHECK-NEXT: "unregistered.empty_dict"() <{}> : () -> ()66// CHECK-NEXT: "unregistered.complex"() <{array = [1, 2, 3], dict = {key1 = 42 : i32, key2 = "value"}, nested = {inner = {deep = 100 : i64}}}> : () -> ()67