mirror of
https://github.com/capstone-engine/llvm-capstone.git
synced 2024-12-14 11:39:35 +00:00
10a80c4413
This patch extends the SubElementAttr interface to allow replacing a contained sub attribute. The attribute that should be replaced is identified by an index which denotes the n-th element returned by the accompanying walkImmediateSubElements method. Using this addition the patch implements replacing SymbolRefAttrs contained within any dialect attributes. Differential Revision: https://reviews.llvm.org/D111357
97 lines
2.9 KiB
MLIR
97 lines
2.9 KiB
MLIR
// RUN: mlir-opt -allow-unregistered-dialect %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 private @symbol_foo() attributes {sym.new_name = "replaced_foo" }
|
|
|
|
// CHECK: func @symbol_bar
|
|
// CHECK: @replaced_foo
|
|
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 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 nested @foo() attributes {sym.new_name = "replaced_foo" }
|
|
}
|
|
}
|
|
|
|
// CHECK: func @symbol_bar
|
|
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 private @failed_repl() attributes {sym.new_name = "replaced_name" }
|
|
|
|
"foo.possibly_unknown_symbol_table"() ({
|
|
}) : () -> ()
|
|
}
|
|
|
|
// -----
|
|
|
|
// Check that replacement works in any implementations of SubElementsAttrInterface
|
|
module {
|
|
// CHECK: func private @replaced_foo
|
|
func private @symbol_foo() attributes {sym.new_name = "replaced_foo" }
|
|
|
|
// CHECK: func @symbol_bar
|
|
func @symbol_bar() {
|
|
// CHECK: foo.op
|
|
// CHECK-SAME: non_symbol_attr,
|
|
// CHECK-SAME: use = [#test.sub_elements_access<[@replaced_foo], @symbol_bar, @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>],
|
|
z_non_symbol_attr_3
|
|
} : () -> ()
|
|
}
|
|
}
|