Skip to content

Commit 81fc54e

Browse files
authored
[PHI]Fix paddle.stack to support big Tensor
1 parent b9dc7c1 commit 81fc54e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddle/phi/infermeta/backward.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1750,7 +1750,7 @@ void StackGradInferMeta(const MetaTensor& out_grad,
17501750
x_grad.size(),
17511751
static_cast<size_t>(dy_dim[axis])));
17521752

1753-
auto vec = common::vectorize<int>(dy_dim);
1753+
auto vec = common::vectorize<int64_t>(dy_dim);
17541754
vec.erase(vec.begin() + axis);
17551755

17561756
for (auto& grad : x_grad) {

0 commit comments

Comments
 (0)