Skip to content

Commit 3b2c4a1

Browse files
authored
[LLM] fix ipc api of big size tensor (#74472) (#74599)
1 parent 7a6ddc9 commit 3b2c4a1

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

paddle/fluid/pybind/tensor.cc

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -792,7 +792,8 @@ 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(
796+
t[4].cast<std::vector<int64_t>>()));
796797

797798
return tensor;
798799
},

0 commit comments

Comments
 (0)