fix share buffer to

PaddlePaddle · Dec 23, 2021 · 57eacfc · 57eacfc · paddle-bot-old · Dec 24, 2021
1 parent 8da9eff
commit 57eacfc
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/paddle/fluid/pybind/imperative.cc b/paddle/fluid/pybind/imperative.cc
@@ -1984,6 +1984,7 @@ void BindImperative(py::module *m_ptr) {
                  platform::errors::InvalidArgument(
                      "Tensor %s has not been initialized!", self->Name()));
              dst_->ShareBufferWith(*src);
+             dst_->ShareDataTypeWith(*src);
            })
       .def("_is_shared_buffer_with",
            [](const std::shared_ptr<imperative::VarBase> &self,