diff options
Diffstat (limited to 'mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp')
| -rw-r--r-- | mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp b/mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp index 775ed8f37344..76df3ecf2d2b 100644 --- a/mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp +++ b/mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp @@ -1908,6 +1908,10 @@ void TransposeOp::getEffects( LogicalResult TransposeOp::fold(FoldAdaptor adaptor, SmallVectorImpl<OpFoldResult> &result) { + // Only the tensor type is supported. + if (!isa<TensorType>(getInput().getType())) + return failure(); + // Single dimension transpose. if (getPermutation().size() == 0) { result.push_back(getInput()); |