From 4e19ba4159f8e498863cd1edb9ecde175edd1f3a Mon Sep 17 00:00:00 2001 From: Alexander Belyaev Date: Thu, 11 Jun 2020 12:40:18 +0200 Subject: [PATCH] [mlir][shape] Add assemblyFormat for `shape.add`. Differential Revision: https://reviews.llvm.org/D81644 --- mlir/include/mlir/Dialect/Shape/IR/ShapeOps.td | 2 ++ mlir/test/Dialect/Shape/ops.mlir | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/mlir/include/mlir/Dialect/Shape/IR/ShapeOps.td b/mlir/include/mlir/Dialect/Shape/IR/ShapeOps.td index 334a15b..78b86bf 100644 --- a/mlir/include/mlir/Dialect/Shape/IR/ShapeOps.td +++ b/mlir/include/mlir/Dialect/Shape/IR/ShapeOps.td @@ -63,6 +63,8 @@ def Shape_AddOp : Shape_Op<"add", [SameOperandsAndResultType]> { let arguments = (ins Shape_SizeType:$lhs, Shape_SizeType:$rhs); let results = (outs Shape_SizeType:$result); + + let assemblyFormat = "$lhs `,` $rhs attr-dict"; } def Shape_BroadcastOp : Shape_Op<"broadcast", []> { diff --git a/mlir/test/Dialect/Shape/ops.mlir b/mlir/test/Dialect/Shape/ops.mlir index a666818..a6718c7 100644 --- a/mlir/test/Dialect/Shape/ops.mlir +++ b/mlir/test/Dialect/Shape/ops.mlir @@ -9,7 +9,7 @@ func @shape_num_elements(%shape : !shape.shape) -> !shape.size { %init = shape.const_size 0 %num_elements = shape.reduce(%shape, %init) -> !shape.size { ^bb0(%index: index, %dim: !shape.size, %lci: !shape.size): - %acc = "shape.add"(%lci, %dim) : (!shape.size, !shape.size) -> !shape.size + %acc = shape.add %lci, %dim shape.yield %acc : !shape.size } return %num_elements : !shape.size -- 2.7.4