fix set data type bug

test=develop
7 years ago · 565b19b7a5
parent 8364688c30
commit 565b19b7a5
1 changed files with 1 additions and 1 deletions
--- 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>();