fix generate shape convert bug (#69561)

phlrain · web-flow · commit 382aa30ae068 · 2024-11-21T10:24:32.000+08:00
diff --git a/paddle/cinn/hlir/dialect/operator/transforms/fuse_shape_ops_into_generate_shape_op_pass.cc b/paddle/cinn/hlir/dialect/operator/transforms/fuse_shape_ops_into_generate_shape_op_pass.cc
@@ -423,6 +423,7 @@ class FuseSingleElementShapeOpsIntoGenerateShapeOpPattern
       auto* user = iter->owner();
       if (IsSingleElementShapeOp(user, &shape_analysis)) return false;
       if (user->isa<cinn::dialect::GenerateShapeOp>()) return false;
+      if (user->isa<pir::ShadowOutputOp>()) return false;
     }
 
     return true;

Original file line number	Diff line number	Diff line change
`@@ -423,6 +423,7 @@ class FuseSingleElementShapeOpsIntoGenerateShapeOpPattern`
`423`	`423`	`auto* user = iter->owner();`
`424`	`424`	`if (IsSingleElementShapeOp(user, &shape_analysis)) return false;`
`425`	`425`	`if (user->isa<cinn::dialect::GenerateShapeOp>()) return false;`
	`426`	`+ if (user->isa<pir::ShadowOutputOp>()) return false;`
`426`	`427`	`}`
`427`	`428`
`428`	`429`	`return true;`