Skip to content

Commit 0a85588

Browse files
authored
1 parent 66b0405 commit 0a85588

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

+1-1
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)