diff options
Diffstat (limited to 'mlir/test/python/dialects/shape.py')
| -rw-r--r-- | mlir/test/python/dialects/shape.py | 5 |
1 files changed, 1 insertions, 4 deletions
diff --git a/mlir/test/python/dialects/shape.py b/mlir/test/python/dialects/shape.py index 1772026cb9b9..7c1c5d6f1cf9 100644 --- a/mlir/test/python/dialects/shape.py +++ b/mlir/test/python/dialects/shape.py @@ -18,15 +18,12 @@ def testConstShape(): with Context() as ctx, Location.unknown(): module = Module.create() f32 = F32Type.get() - indexT = IndexType.get() with InsertionPoint(module.body): @builtin.FuncOp.from_py_func( RankedTensorType.get((12, -1), f32)) def const_shape_tensor(arg): - return shape.ConstShapeOp(RankedTensorType.get((2,), indexT), - DenseElementsAttr.get(np.array([10, 20]))) + return shape.ConstShapeOp(DenseElementsAttr.get(np.array([10, 20]))) # CHECK-LABEL: func @const_shape_tensor(%arg0: tensor<12x?xf32>) # CHECK: shape.const_shape [10, 20] : tensor<2xindex> print(module) - |