We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9863408 commit 6320b01Copy full SHA for 6320b01
paddle/cinn/hlir/dialect/operator/transforms/fold_full_pass.cc
@@ -49,7 +49,7 @@ class FoldFullWithReshapeOpPattern : public pir::OpRewritePattern<OPTYPE> {
49
phi::IntArray(out_shape.Get(), out_shape.size()));
50
51
auto new_full_op = rewriter.Build<paddle::dialect::FullOp>(attrs);
52
-
+ new_full_op->result(0).set_type(op->result(0).type());
53
rewriter.ReplaceAllUsesWith(op->result(0), new_full_op->result(0));
54
rewriter.EraseOp(op);
55
if (pre_op->use_empty()) {
0 commit comments