diff --git a/paddle/cinn/hlir/dialect/operator/transforms/pd_to_cinn_pass.cc b/paddle/cinn/hlir/dialect/operator/transforms/pd_to_cinn_pass.cc index 8c2becde5d990d..e536487e771008 100644 --- a/paddle/cinn/hlir/dialect/operator/transforms/pd_to_cinn_pass.cc +++ b/paddle/cinn/hlir/dialect/operator/transforms/pd_to_cinn_pass.cc @@ -337,6 +337,9 @@ class SliceOpPattern : public pir::OpRewritePattern { end_vec, infer_flags, decrease_axis); + // NOTE(Aurelius84): In SliceRawInferMeta, it not always share_lod, so + // we need to update it maually. + cinn_slice.result(0).set_type(op.result(0).type()); rewriter.ReplaceAllUsesWith(op.result(0), cinn_slice.result(0)); rewriter.EraseOp(op);