[Mlir-commits] [mlir] 4bd68f2 - Add NoSideEffect trait to shape.split_at and shape.concat

[Jacques Pienaar]

Author: Jing Pu
Date: 2021-02-09T15:19:53-08:00
New Revision: 4bd68f238cf28413f6ad1d24fd5de1e02ff06d33

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

LOG: Add NoSideEffect trait to shape.split_at and shape.concat

Reviewed By: jpienaar, silvas

Differential Revision: https://reviews.llvm.org/D96358

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 0cbb910e062c..ba89a9455781 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`