0e06ec5961
"transform.structured.pad" now returns all `tensor::PadOp` in addition to the padded ops. Also add a test case that shows how to force an allocation for "tensor.pad" ops with a custom memory space. Differential Revision: https://reviews.llvm.org/D153555
50 lines
2.2 KiB
MLIR
50 lines
2.2 KiB
MLIR
// RUN: mlir-opt %s | mlir-opt | FileCheck %s
|
|
|
|
transform.sequence failures(propagate) {
|
|
^bb1(%arg0: !transform.any_op):
|
|
// CHECK %{{.*}}, %{{.*}}:2 = transform.structured.tile
|
|
%0, %1:2 = transform.structured.tile %arg0 [2, 0, 3] : (!transform.any_op) -> (!transform.any_op, !transform.any_op, !transform.any_op)
|
|
}
|
|
|
|
transform.sequence failures(propagate) {
|
|
^bb1(%arg0: !transform.any_op):
|
|
%0:2 = transform.structured.split %arg0 after 42 { dimension = 0 } : !transform.any_op
|
|
transform.structured.split %0#0 after %0#1 { dimension = 1 } : !transform.any_op, !transform.any_op
|
|
}
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Check that operations are registered correctly through the extension
|
|
// mechanism. Their syntax is generated and requires no additional testing since
|
|
// we test the generator.
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
transform.sequence failures(propagate) {
|
|
^bb1(%arg0: !transform.any_op):
|
|
// CHECK: transform.structured.pad
|
|
%0, %1 = transform.structured.pad %arg0 : (!transform.any_op) -> (!transform.any_op, !transform.any_op)
|
|
}
|
|
|
|
transform.sequence failures(propagate) {
|
|
^bb1(%arg0: !transform.any_op):
|
|
// CHECK: transform.structured.interchange
|
|
%0 = transform.structured.interchange %arg0 : (!transform.any_op) -> !transform.any_op
|
|
}
|
|
|
|
transform.sequence failures(propagate) {
|
|
^bb1(%arg0: !transform.any_op):
|
|
// CHECK: transform.structured.scalarize
|
|
%0 = transform.structured.scalarize %arg0 : (!transform.any_op) -> !transform.any_op
|
|
}
|
|
|
|
// Check that the second argument of `fuse_into_containing_op` is not consumed
|
|
// (if it had been, we would have seen a diagnostic about multiple consumers).
|
|
transform.sequence failures(propagate) {
|
|
^bb1(%arg0: !transform.any_op, %arg1: !transform.any_op, %arg2: !transform.any_op):
|
|
%loop = transform.structured.match ops{["scf.forall"]} in %arg0
|
|
: (!transform.any_op) -> !transform.any_op
|
|
%0:2 = transform.structured.fuse_into_containing_op %arg1 into %loop
|
|
: (!transform.any_op, !transform.any_op) -> (!transform.any_op, !transform.any_op)
|
|
%1:2 = transform.structured.fuse_into_containing_op %arg2 into %loop
|
|
: (!transform.any_op, !transform.any_op) -> (!transform.any_op, !transform.any_op)
|
|
}
|