Fix: exclude inPlaceOp in groupParallel

Change-Id: I71cb8b01975c9bd8cd945314bff758110fdfeabc
sophgo · Jan 28, 2024 · dc061a7 · dc061a7
1 parent fc934d2
commit dc061a7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/Dialect/Tpu/Transforms/CoreParallel/CoreMatch.cpp b/lib/Dialect/Tpu/Transforms/CoreParallel/CoreMatch.cpp
@@ -321,7 +321,8 @@ struct CommonMatch : public RewritePattern {
       auto users = value.getUsers();
       for (auto left = users.begin(); left != users.end(); left++) {
         auto left_op = *left;
-        if (isa<tpu::ReshapeOp>(left_op)) {
+        // inPlace op
+        if (isa<tpu::ReshapeOp, tpu::SliceOp, tpu::ConcatOp>(left_op)) {
           continue;
         }
         if (find_f(same_ops, left_op)) {