[Mlir-commits] [mlir] f1ab755 - [MLIR] Fix operand type in `from_extent_tensor` in the shape dialect

[Frederik Gossen]

Author: Frederik Gossen
Date: 2020-05-28T14:12:47Z
New Revision: f1ab7550bcd51c353a1cac0303df9bbe960b7eab

URL: https://github.com/llvm/llvm-project/commit/f1ab7550bcd51c353a1cac0303df9bbe960b7eab
DIFF: https://github.com/llvm/llvm-project/commit/f1ab7550bcd51c353a1cac0303df9bbe960b7eab.diff

LOG: [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

Added: 
    

Modified: 
    mlir/include/mlir/Dialect/Shape/IR/ShapeOps.td

Removed: 
    


################################################################################
diff  --git a/mlir/include/mlir/Dialect/Shape/IR/ShapeOps.td b/mlir/include/mlir/Dialect/Shape/IR/ShapeOps.td
index 0d300d3c64c8..460f5becc1f9 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);
 }