Skip to content

Commit 4d052ff

Browse files
Update ToTensorOp::getType() to return TensorLikeType
1 parent 2de5728 commit 4d052ff

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

mlir/include/mlir/Dialect/Bufferization/IR/BufferizationOps.td

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -461,11 +461,8 @@ def Bufferization_ToTensorOp : Bufferization_Op<"to_tensor", [
461461

462462
let extraClassDeclaration = [{
463463
/// The result of a to_tensor is always a tensor.
464-
TensorType getType() {
465-
Type resultType = getResult().getType();
466-
if (::llvm::isa<TensorType>(resultType))
467-
return ::llvm::cast<TensorType>(resultType);
468-
return {};
464+
::mlir::bufferization::TensorLikeType getType() {
465+
return getResult().getType();
469466
}
470467

471468
//===------------------------------------------------------------------===//

0 commit comments

Comments
 (0)