diff --git a/mlir/lib/Conversion/TosaToTensor/TosaToTensor.cpp b/mlir/lib/Conversion/TosaToTensor/TosaToTensor.cpp index 7a62429d37004..06ec53d19b1e9 100644 --- a/mlir/lib/Conversion/TosaToTensor/TosaToTensor.cpp +++ b/mlir/lib/Conversion/TosaToTensor/TosaToTensor.cpp @@ -244,8 +244,7 @@ class SliceConverter : public OpConversionPattern { Location loc = sliceOp.getLoc(); Value input = adaptor.getInput(); ShapedType resultType = cast(sliceOp.getType()); - if (llvm::isa(resultType) || - resultType.getRank() != static_cast(sliceOp.getSize().size())) + if (llvm::isa(resultType)) return failure(); SmallVector strides, sizes; ArrayRef starts = sliceOp.getStart(); diff --git a/mlir/test/Conversion/TosaToTensor/tosa-to-tensor-invalid.mlir b/mlir/test/Conversion/TosaToTensor/tosa-to-tensor-invalid.mlir index cfa1c41f24679..36eb4d4669b07 100644 --- a/mlir/test/Conversion/TosaToTensor/tosa-to-tensor-invalid.mlir +++ b/mlir/test/Conversion/TosaToTensor/tosa-to-tensor-invalid.mlir @@ -6,10 +6,3 @@ func.func @slice_resultType_unranked(%arg0: tensor) -> (tensor<*xf32>) { %0 = "tosa.slice"(%arg0) {start = array, size = array} : (tensor) -> (tensor<*xf32>) return %0 : tensor<*xf32> } - -// CHECK-LABEL: @slice_resultRank_neq_opSize -func.func @slice_resultRank_neq_opSize(%arg0: tensor<12xf32>) -> (tensor<2xf32>) { - // expected-error@+1 {{failed to legalize operation 'tosa.slice'}} - %0 = "tosa.slice"(%arg0) {start = array, size = array} : (tensor<12xf32>) -> (tensor<2xf32>) - return %0 : tensor<2xf32> -}