We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent e7f1536 commit 42dbba1Copy full SHA for 42dbba1
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