aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJing Pu <jingpu@google.com>2021-02-09 15:19:13 -0800
committerJacques Pienaar <jpienaar@google.com>2021-02-09 15:19:53 -0800
commit4bd68f238cf28413f6ad1d24fd5de1e02ff06d33 (patch)
treedb72c0ebc6dba88c741a652042b82e37b5084c19
parente090182fe153c9ceea50b1807f8ca5c13729e402 (diff)
downloadllvm-4bd68f238cf28413f6ad1d24fd5de1e02ff06d33.zip
llvm-4bd68f238cf28413f6ad1d24fd5de1e02ff06d33.tar.gz
llvm-4bd68f238cf28413f6ad1d24fd5de1e02ff06d33.tar.bz2
Add NoSideEffect trait to shape.split_at and shape.concat
Reviewed By: jpienaar, silvas Differential Revision: https://reviews.llvm.org/D96358
-rw-r--r--mlir/include/mlir/Dialect/Shape/IR/ShapeOps.td4
1 files changed, 2 insertions, 2 deletions
diff --git a/mlir/include/mlir/Dialect/Shape/IR/ShapeOps.td b/mlir/include/mlir/Dialect/Shape/IR/ShapeOps.td
index 0cbb910..ba89a94 100644
--- a/mlir/include/mlir/Dialect/Shape/IR/ShapeOps.td
+++ b/mlir/include/mlir/Dialect/Shape/IR/ShapeOps.td
@@ -526,7 +526,7 @@ def Shape_DebugPrintOp : Shape_Op<"debug_print", []> {
let results = (outs Shape_ShapeOrSizeType:$output);
}
-def Shape_SplitAtOp : Shape_Op<"split_at", []> {
+def Shape_SplitAtOp : Shape_Op<"split_at", [NoSideEffect]> {
let summary = "Splits a shape at a given index";
let description = [{
Splits a shape at a given dimension `index`, returning two shapes.
@@ -555,7 +555,7 @@ def Shape_SplitAtOp : Shape_Op<"split_at", []> {
let hasFolder = 1;
}
-def Shape_ConcatOp : Shape_Op<"concat", []> {
+def Shape_ConcatOp : Shape_Op<"concat", [NoSideEffect]> {
let summary = "Concatenates two shapes";
let description = [{
Creates a shape whose dimensions consist of first the dimensions from `lhs`