Skip to content

Commit 42dbba1

Browse files
authored
fix bug (#71562)
1 parent e7f1536 commit 42dbba1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddle/cinn/hlir/dialect/operator/transforms/fold_full_pass.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ class FoldFullWithReshapeOpPattern : public pir::OpRewritePattern<OPTYPE> {
4949
phi::IntArray(out_shape.Get(), out_shape.size()));
5050

5151
auto new_full_op = rewriter.Build<paddle::dialect::FullOp>(attrs);
52-
52+
new_full_op->result(0).set_type(op->result(0).type());
5353
rewriter.ReplaceAllUsesWith(op->result(0), new_full_op->result(0));
5454
rewriter.EraseOp(op);
5555
if (pre_op->use_empty()) {

0 commit comments

Comments
 (0)