fix set data type bug

velconia · velconia · commit 565b19b7a506 · 2019-03-18T20:39:41.000+08:00
test=develop
diff --git 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>();

Original file line number	Diff line number	Diff line change
`@@ -195,7 +195,7 @@ class VarBase {`
`195`	`195`	`// data type. e.g.. FP32`
`196`	`196`	`inline void SetDataType(framework::proto::VarType::Type type) {`
`197`	`197`	`auto tensor = var_->GetMutable<framework::LoDTensor>();`
`198`		`- tensor->mutable_data(place_, type);`
	`198`	`+ tensor->mutable_data(tensor->place(), type);`
`199`	`199`	`}`
`200`	`200`	`inline framework::proto::VarType::Type DataType() const {`
`201`	`201`	`auto tensor = var_->Get<framework::LoDTensor>();`