From 24c0a8256f4425f5aeedf6ea5183ca1ac3e2d076 Mon Sep 17 00:00:00 2001
From: lvliang <lvliang18@huawei.com>
Date: Sun, 28 Jun 2020 15:23:41 +0800
Subject: [PATCH] fix-grad-value-is-wrong-in-pynative-hook

---
 mindspore/ccsrc/pynative/pynative_execute.cc | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/mindspore/ccsrc/pynative/pynative_execute.cc b/mindspore/ccsrc/pynative/pynative_execute.cc
index d65807b9f6..b5f0958a1a 100644
--- a/mindspore/ccsrc/pynative/pynative_execute.cc
+++ b/mindspore/ccsrc/pynative/pynative_execute.cc
@@ -261,8 +261,10 @@ py::object RunOpInVM(const OpExecInfoPtr &op_exec_info, PynativeStatusCode *stat
       if (py::hasattr(input, "__parameter__")) {
         result[i] = py::getattr(input, "data");
       } else {
-        auto tensor = py::cast<tensor::TensorPtr>(op_inputs[i]);
+        auto tensor = py::cast<tensor::TensorPtr>(input);
         auto new_tensor = std::make_shared<tensor::Tensor>(tensor->data());
+        new_tensor->set_device_address(tensor->device_address());
+        new_tensor->set_dirty(tensor->is_dirty());
         result[i] = new_tensor;
       }
     }