bolt/deps/llvm-18.1.8/mlir/test/Dialect/Transform/transform-state-extension.mlir
2025-02-14 19:21:04 +01:00

89 lines
3.1 KiB
MLIR

// RUN: mlir-opt %s -test-transform-dialect-interpreter -verify-diagnostics -split-input-file
// expected-note @below {{associated payload op}}
module {
transform.sequence failures(propagate) {
^bb0(%arg0: !transform.any_op):
// expected-remark @below {{extension absent}}
test_check_if_test_extension_present %arg0 : !transform.any_op
test_add_test_extension "A"
// expected-remark @below {{extension present, A}}
test_check_if_test_extension_present %arg0 : !transform.any_op
test_remove_test_extension
// expected-remark @below {{extension absent}}
test_check_if_test_extension_present %arg0 : !transform.any_op
}
}
// -----
// expected-note @below {{associated payload op}}
module {
transform.sequence failures(propagate) {
^bb0(%arg0: !transform.any_op):
test_add_test_extension "A"
test_remove_test_extension
test_add_test_extension "B"
// expected-remark @below {{extension present, B}}
test_check_if_test_extension_present %arg0 : !transform.any_op
}
}
// -----
// expected-note @below {{associated payload op}}
module {
transform.sequence failures(propagate) {
^bb0(%arg0: !transform.any_op):
test_add_test_extension "A"
// expected-remark @below {{extension present, A}}
test_check_if_test_extension_present %arg0 : !transform.any_op
// expected-note @below {{associated payload op}}
test_remap_operand_to_self %arg0 : (!transform.any_op) -> !transform.any_op
// expected-remark @below {{extension present, A}}
test_check_if_test_extension_present %arg0 : !transform.any_op
}
}
// -----
transform.sequence failures(propagate) {
^bb0(%arg0: !transform.any_op):
test_add_test_extension "A"
// This is okay because we are replacing the top-level module operation
// (0 results) with this operation that has _more_ (1) results.
%dummy = test_remap_operand_to_self %arg0 : (!transform.any_op) -> !transform.any_op
}
// -----
transform.sequence failures(propagate) {
^bb0(%arg0: !transform.any_op):
test_add_test_extension "A"
%dummy = test_remap_operand_to_self %arg0 : (!transform.any_op) -> !transform.any_op
// This is still okay. Even though we are replacing the previous
// operation with (1 result) with this operation that has less (0) results,
// there is no handle to the result, hence no issue with value handle update.
test_remap_operand_to_self %dummy : (!transform.any_op) -> !transform.any_op
}
// -----
transform.sequence failures(propagate) {
^bb0(%arg0: !transform.any_op):
test_add_test_extension "A"
// expected-error @below {{cannot replace an op with another op producing fewer results while tracking handles}}
%dummy = test_remap_operand_to_self %arg0 : (!transform.any_op) -> !transform.any_op
%valuehandle = transform.get_result %dummy[0] : (!transform.any_op) -> !transform.any_value
test_remap_operand_to_self %dummy : (!transform.any_op) -> ()
}
// -----
module {
transform.sequence failures(suppress) {
^bb0(%arg0: !transform.any_op):
// expected-error @below {{TestTransformStateExtension missing}}
test_remap_operand_to_self %arg0 : (!transform.any_op) -> !transform.any_op
}
}