add code

paddle/fluid/framework/dlpack_tensor.cc

@@ -188,6 +188,7 @@ DLPackTensor::DLPackTensor(phi::DenseTensor &tensor, LaneType lanes) {
 
   auto tensor = new DLManagedTensor;
   tensor->dl_tensor = t_;
+  tensor->manager_ctx = dt_;
 
   tensor->deleter = [](DLManagedTensor *arg) {
     phi::DenseTensor *tensor_ptr =
@@ -200,8 +201,6 @@ DLPackTensor::DLPackTensor(phi::DenseTensor &tensor, LaneType lanes) {
     delete arg;
   };
 
-  tensor->manager_ctx = dt_;
-
   return tensor;
 }
 

paddle/fluid/framework/dlpack_tensor.h

@@ -42,7 +42,6 @@ class DLPackTensor {
   // The shape in DLTensor is defined as int64_t*
   // Add this member to make TVMTensor init without heap allocation
   ShapeType shape_[DDim::kMaxRank];
-
   phi::DenseTensor* dt_;
 };
 

paddle/fluid/pybind/tensor.cc

@@ -473,9 +473,9 @@ void BindTensor(pybind11::module &m) {  // NOLINT
            )DOC")
       .def("_to_dlpack",
            [](phi::DenseTensor &self) {
-             DLPackTensor dlpack_tensor(self, 1);
              pybind11::handle tensor_handle = pybind11::cast(&self);
              tensor_handle.inc_ref();
+             DLPackTensor dlpack_tensor(self, 1);
              DLManagedTensor *dmt = dlpack_tensor.ToDLManagedTensor();
              auto capsule = pybind11::capsule(
                  static_cast<void *>(dmt), "dltensor", [](PyObject *ptr) {