Skip to content

Commit 8a93404

Browse files
fix_split_infermeta (#49745)
1 parent 6d2d8e5 commit 8a93404

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

paddle/phi/infermeta/unary.cc

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3184,11 +3184,11 @@ void FillSplitOutDims(const MetaTensor& x,
31843184
(*out)[i]->set_dtype(x.dtype());
31853185
(*out)[i]->set_dims(out_dims[i]);
31863186
(*out)[i]->set_layout(x.layout());
3187+
(*out)[i]->share_lod(x);
31873188
} else {
31883189
(*out)[i]->set_dtype(x.dtype());
31893190
(*out)[i]->set_dims(out_dims[i]);
31903191
(*out)[i]->set_layout(x.layout());
3191-
(*out)[i]->share_lod(x);
31923192
}
31933193
}
31943194
}
@@ -3219,11 +3219,11 @@ void SplitInferMeta(const MetaTensor& x,
32193219
out[i]->set_dtype(x.dtype());
32203220
out[i]->set_dims(out_dims[i]);
32213221
out[i]->set_layout(x.layout());
3222+
out[i]->share_lod(x);
32223223
} else {
32233224
out[i]->set_dtype(x.dtype());
32243225
out[i]->set_dims(out_dims[i]);
32253226
out[i]->set_layout(x.layout());
3226-
out[i]->share_lod(x);
32273227
}
32283228
}
32293229
} else {
@@ -3310,11 +3310,11 @@ void SplitWithNumInferMeta(const MetaTensor& x,
33103310
out[i]->set_dtype(x.dtype());
33113311
out[i]->set_dims(out_dims[i]);
33123312
out[i]->set_layout(x.layout());
3313+
out[i]->share_lod(x);
33133314
} else {
33143315
out[i]->set_dtype(x.dtype());
33153316
out[i]->set_dims(out_dims[i]);
33163317
out[i]->set_layout(x.layout());
3317-
out[i]->share_lod(x);
33183318
}
33193319
}
33203320
} else {

0 commit comments

Comments
 (0)