From 565b19b7a50698894cc238b375dd73547548958c Mon Sep 17 00:00:00 2001
From: minqiyang <minqiyang@baidu.com>
Date: Mon, 18 Mar 2019 20:39:41 +0800
Subject: [PATCH] fix set data type bug

test=develop
---
 paddle/fluid/imperative/layer.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddle/fluid/imperative/layer.h b/paddle/fluid/imperative/layer.h
index f210cd17453..f409190a70b 100644
--- a/paddle/fluid/imperative/layer.h
+++ b/paddle/fluid/imperative/layer.h
@@ -195,7 +195,7 @@ class VarBase {
   // data type. e.g.. FP32
   inline void SetDataType(framework::proto::VarType::Type type) {
     auto tensor = var_->GetMutable<framework::LoDTensor>();
-    tensor->mutable_data(place_, type);
+    tensor->mutable_data(tensor->place(), type);
   }
   inline framework::proto::VarType::Type DataType() const {
     auto tensor = var_->Get<framework::LoDTensor>();
-- 
GitLab