From 146861f605e197ec872249b1b20609a35d2c6edf Mon Sep 17 00:00:00 2001
From: chujinjin <chujinjin52@huawei.com>
Date: Wed, 10 Jun 2020 14:38:20 +0800
Subject: [PATCH] fix loadinputdata for pynative

---
 mindspore/ccsrc/session/session_basic.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mindspore/ccsrc/session/session_basic.cc b/mindspore/ccsrc/session/session_basic.cc
index d11446a8ba..1d7e8da5b0 100644
--- a/mindspore/ccsrc/session/session_basic.cc
+++ b/mindspore/ccsrc/session/session_basic.cc
@@ -686,7 +686,7 @@ void SessionBasic::LoadInputData(const std::shared_ptr<KernelGraph> &kernel_grap
         }
       }
       if (need_sync) {
-        if (AnfAlgo::IsParameterWeight(pk_node)) {
+        if (ms_context->execution_mode() == kPynativeMode || AnfAlgo::IsParameterWeight(pk_node)) {
           tensor->set_device_address(device_address);
         }
         MS_EXCEPTION_IF_NULL(device_address);