Skip to content

Commit e6f68e7

Browse files
authored
Fix SliceTensor to fix paddle.functional.diff for big tensor (#74451)
1 parent 4f31055 commit e6f68e7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddle/phi/kernels/funcs/broadcast_function.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -584,7 +584,7 @@ static void SliceTensor(DenseTensor *x,
584584
DenseTensorMeta meta(share->dtype(),
585585
new_dim,
586586
share->layout(),
587-
offset * SizeOf(share->dtype()));
587+
offset * SizeOf(share->dtype()) + share->offset());
588588
x->set_meta(meta);
589589
x->ShareBufferWith(*(share), true);
590590
x->Resize(new_dim);

0 commit comments

Comments
 (0)