1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
// RUN: mlir-opt -allow-unregistered-dialect -mlir-print-local-scope %s -test-symbol-rauw -split-input-file | FileCheck %s
// Symbol references to the module itself don't affect uses of symbols within
// its table.
// CHECK: module
// CHECK-SAME: @symbol_foo
module attributes {sym.outside_use = @symbol_foo } {
// CHECK: func private @replaced_foo
func.func private @symbol_foo() attributes {sym.new_name = "replaced_foo" }
// CHECK: func @symbol_bar
// CHECK: @replaced_foo
func.func @symbol_bar() attributes {sym.use = @symbol_foo} {
// CHECK: foo.op
// CHECK-SAME: non_symbol_attr,
// CHECK-SAME: use = [{nested_symbol = [@replaced_foo], other_use = @symbol_bar, z_use = @replaced_foo}],
// CHECK-SAME: z_non_symbol_attr_3
"foo.op"() {
non_symbol_attr,
use = [{nested_symbol = [@symbol_foo], other_use = @symbol_bar, z_use = @symbol_foo}],
z_non_symbol_attr_3
} : () -> ()
}
// CHECK: module attributes {test.reference = @replaced_foo}
module attributes {test.reference = @symbol_foo} {
// CHECK: foo.op
// CHECK-SAME: @symbol_foo
"foo.op"() {test.nested_reference = @symbol_foo} : () -> ()
}
}
// -----
// Check the support for nested references.
// CHECK: module
module {
// CHECK: module @module_a
module @module_a {
// CHECK: func nested @replaced_foo
func.func nested @foo() attributes {sym.new_name = "replaced_foo" }
}
// CHECK: module @replaced_module_b
module @module_b attributes {sym.new_name = "replaced_module_b"} {
// CHECK: module @replaced_module_c
module @module_c attributes {sym.new_name = "replaced_module_c"} {
// CHECK: func nested @replaced_foo
func.func nested @foo() attributes {sym.new_name = "replaced_foo" }
}
}
// CHECK: func @symbol_bar
func.func @symbol_bar() {
// CHECK: foo.op
// CHECK-SAME: use_1 = @module_a::@replaced_foo
// CHECK-SAME: use_2 = @replaced_module_b::@replaced_module_c::@replaced_foo
"foo.op"() {
use_1 = @module_a::@foo,
use_2 = @module_b::@module_c::@foo
} : () -> ()
}
}
// -----
// Check that the replacement fails for potentially unknown symbol tables.
module {
// CHECK: func private @failed_repl
func.func private @failed_repl() attributes {sym.new_name = "replaced_name" }
"foo.possibly_unknown_symbol_table"() ({
}) : () -> ()
}
// -----
// Check that replacement works in any implementations of SubElements.
module {
// CHECK: func private @replaced_foo
func.func private @symbol_foo() attributes {sym.new_name = "replaced_foo" }
// CHECK: func @symbol_bar
func.func @symbol_bar() {
// CHECK: foo.op
// CHECK-SAME: non_symbol_attr,
// CHECK-SAME: use = [#test.sub_elements_access<[@replaced_foo], @symbol_bar, @replaced_foo>, distinct[0]<@replaced_foo>],
// CHECK-SAME: z_non_symbol_attr_3
"foo.op"() {
non_symbol_attr,
use = [#test.sub_elements_access<[@symbol_foo],@symbol_bar,@symbol_foo>, distinct[0]<@symbol_foo>],
z_non_symbol_attr_3
} : () -> ()
}
}
// -----
module {
// CHECK: module @replaced_foo
module @foo attributes {sym.new_name = "replaced_foo" } {
// CHECK: func.func private @foo
func.func private @foo()
}
// CHECK: foo.op
// CHECK-SAME: use = @replaced_foo::@foo
"foo.op"() {
use = @foo::@foo
} : () -> ()
}
|
