Add test for the unranked case

Author: krzysz00

mlir/test/Dialect/MemRef/int-range-inference.mlir

@@ -59,3 +59,16 @@ func.func @dim_some_omitting_dynamic(%m: memref<?x3x5xi32>, %x: index) -> index
   %2 = test.reflect_bounds %1 : index
   return %2 : index
 }
+
+// -----
+
+// CHECK-LABEL: @dim_unranked
+// CHECK: %[[op:.+]] = memref.dim
+// CHECK: %[[ret:.+]] = test.reflect_bounds {smax = 9223372036854775807 : index, smin = 0 : index, umax = 9223372036854775807 : index, umin = 0 : index} %[[op]]
+// CHECK: return %[[ret]]
+func.func @dim_unranked(%m: memref<*xi32>) -> index {
+  %c0 = arith.constant 0 : index
+  %0 = memref.dim %m, %c0 : memref<*xi32>
+  %1 = test.reflect_bounds %0 : index
+  return %1 : index
+}

mlir/test/Dialect/Tensor/int-range-inference.mlir

@@ -3,9 +3,9 @@
 // CHECK-LABEL: @dim_const
 // CHECK: %[[ret:.+]] = arith.constant 3 : index
 // CHECK: return %[[ret]]
-func.func @dim_const(%m: tensor<3x5xi32>) -> index {
+func.func @dim_const(%t: tensor<3x5xi32>) -> index {
   %c0 = arith.constant 0 : index
-  %0 = tensor.dim %m, %c0 : tensor<3x5xi32>
+  %0 = tensor.dim %t, %c0 : tensor<3x5xi32>
   return %0 : index
 }
 
@@ -15,8 +15,8 @@ func.func @dim_const(%m: tensor<3x5xi32>) -> index {
 // CHECK: %[[op:.+]] = tensor.dim
 // CHECK: %[[ret:.+]] = test.reflect_bounds {smax = 5 : index, smin = 3 : index, umax = 5 : index, umin = 3 : index} %[[op]]
 // CHECK: return %[[ret]]
-func.func @dim_any_static(%m: tensor<3x5xi32>, %x: index) -> index {
-  %0 = tensor.dim %m, %x : tensor<3x5xi32>
+func.func @dim_any_static(%t: tensor<3x5xi32>, %x: index) -> index {
+  %0 = tensor.dim %t, %x : tensor<3x5xi32>
   %1 = test.reflect_bounds %0 : index
   return %1 : index
 }
@@ -27,9 +27,9 @@ func.func @dim_any_static(%m: tensor<3x5xi32>, %x: index) -> index {
 // CHECK: %[[op:.+]] = tensor.dim
 // CHECK: %[[ret:.+]] = test.reflect_bounds {smax = 9223372036854775807 : index, smin = 0 : index, umax = 9223372036854775807 : index, umin = 0 : index} %[[op]]
 // CHECK: return %[[ret]]
-func.func @dim_dynamic(%m: tensor<?x5xi32>) -> index {
+func.func @dim_dynamic(%t: tensor<?x5xi32>) -> index {
   %c0 = arith.constant 0 : index
-  %0 = tensor.dim %m, %c0 : tensor<?x5xi32>
+  %0 = tensor.dim %t, %c0 : tensor<?x5xi32>
   %1 = test.reflect_bounds %0 : index
   return %1 : index
 }
@@ -40,8 +40,8 @@ func.func @dim_dynamic(%m: tensor<?x5xi32>) -> index {
 // CHECK: %[[op:.+]] = tensor.dim
 // CHECK: %[[ret:.+]] = test.reflect_bounds {smax = 9223372036854775807 : index, smin = 0 : index, umax = 9223372036854775807 : index, umin = 0 : index} %[[op]]
 // CHECK: return %[[ret]]
-func.func @dim_any_dynamic(%m: tensor<?x5xi32>, %x: index) -> index {
-  %0 = tensor.dim %m, %x : tensor<?x5xi32>
+func.func @dim_any_dynamic(%t: tensor<?x5xi32>, %x: index) -> index {
+  %0 = tensor.dim %t, %x : tensor<?x5xi32>
   %1 = test.reflect_bounds %0 : index
   return %1 : index
 }
@@ -52,10 +52,23 @@ func.func @dim_any_dynamic(%m: tensor<?x5xi32>, %x: index) -> index {
 // CHECK: %[[op:.+]] = tensor.dim
 // CHECK: %[[ret:.+]] = test.reflect_bounds {smax = 5 : index, smin = 3 : index, umax = 5 : index, umin = 3 : index} %[[op]]
 // CHECK: return %[[ret]]
-func.func @dim_some_omitting_dynamic(%m: tensor<?x3x5xi32>, %x: index) -> index {
+func.func @dim_some_omitting_dynamic(%t: tensor<?x3x5xi32>, %x: index) -> index {
   %c1 = arith.constant 1 : index
   %0 = arith.maxsi %x, %c1 : index
-  %1 = tensor.dim %m, %0 : tensor<?x3x5xi32>
+  %1 = tensor.dim %t, %0 : tensor<?x3x5xi32>
   %2 = test.reflect_bounds %1 : index
   return %2 : index
 }
+
+// -----
+
+// CHECK-LABEL: @dim_unranked
+// CHECK: %[[op:.+]] = tensor.dim
+// CHECK: %[[ret:.+]] = test.reflect_bounds {smax = 9223372036854775807 : index, smin = 0 : index, umax = 9223372036854775807 : index, umin = 0 : index} %[[op]]
+// CHECK: return %[[ret]]
+func.func @dim_unranked(%t: tensor<*xi32>) -> index {
+  %c0 = arith.constant 0 : index
+  %0 = tensor.dim %t, %c0 : tensor<*xi32>
+  %1 = test.reflect_bounds %0 : index
+  return %1 : index
+}