Skip to content

Commit 628e6be

Browse files
committed
[LLM] fix ipc api of big size tensor (PaddlePaddle#74472)
1 parent f1ae790 commit 628e6be

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddle/fluid/pybind/tensor.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -792,7 +792,7 @@ void BindTensor(pybind11::module &m) { // NOLINT
792792
tensor.ResetHolderWithType(
793793
shared_reader_holder,
794794
static_cast<phi::DataType>(t[3].cast<int>()));
795-
tensor.Resize(common::make_ddim(t[4].cast<std::vector<int>>()));
795+
tensor.Resize(common::make_ddim(t[4].cast<std::vector<int64_t>>()));
796796

797797
return tensor;
798798
},

0 commit comments

Comments
 (0)