From f1ab7550bcd51c353a1cac0303df9bbe960b7eab Mon Sep 17 00:00:00 2001 From: Frederik Gossen Date: Thu, 28 May 2020 14:10:11 +0000 Subject: [PATCH] [MLIR] Fix operand type in `from_extent_tensor` in the shape dialect The operand of `from_extent_tensor` is now of the same index type as the result type of the inverse operation `to_extent_tensor`. Differential Revision: https://reviews.llvm.org/D80283 --- mlir/include/mlir/Dialect/Shape/IR/ShapeOps.td | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mlir/include/mlir/Dialect/Shape/IR/ShapeOps.td b/mlir/include/mlir/Dialect/Shape/IR/ShapeOps.td index 0d300d3c64c..460f5becc1f 100644 --- a/mlir/include/mlir/Dialect/Shape/IR/ShapeOps.td +++ b/mlir/include/mlir/Dialect/Shape/IR/ShapeOps.td @@ -163,7 +163,7 @@ def Shape_FromExtentTensorOp : Shape_Op<"from_extent_tensor", []> { extents match the values of the elements. }]; - let arguments = (ins I32Tensor:$input); + let arguments = (ins IndexTensor:$input); let results = (outs Shape_ShapeType:$result); } -- 2.11.0