Skip to content

Commit 9bc81d7

Browse files
committed
Add test case for unranked type
1 parent a436c01 commit 9bc81d7

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

mlir/test/Dialect/Math/expand-math.mlir

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -774,3 +774,16 @@ func.func @non_static_shape_ceil_op(%arg: tensor<?xf32>) -> tensor<?xf32>{
774774
%a = math.ceil %arg : tensor<?xf32>
775775
return %a: tensor<?xf32>
776776
}
777+
778+
// -----
779+
780+
// CHECK-LABEL: func.func @unranked_ceil_op
781+
// CHECK-SAME: (%[[ARG:.*]]: tensor<*xf32>)
782+
// CHECK-SAME: -> tensor<*xf32>
783+
// CHECK: %[[CEIL:.*]] = math.ceil %[[ARG]] : tensor<*xf32>
784+
// CHECK: return %[[CEIL]] : tensor<*xf32>
785+
786+
func.func @unranked_ceil_op(%arg: tensor<*xf32>) -> tensor<*xf32>{
787+
%a = math.ceil %arg : tensor<*xf32>
788+
return %a: tensor<*xf32>
789+
}

0 commit comments

Comments
 (0)