scope.cc 3.3 KB
Newer Older
Y
Yi Wang 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
/* Copyright (c) 2016 PaddlePaddle Authors. All Rights Reserve.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License. */

#include "paddle/framework/scope.h"
Y
Yang Yang 已提交
16

Q
qijun 已提交
17 18
#include <memory>  // for unique_ptr
#include <mutex>   // for call_once
19
#include "glog/logging.h"
Y
Yi Wang 已提交
20
#include "paddle/string/printf.h"
Y
Yi Wang 已提交
21 22 23 24 25

namespace paddle {
namespace framework {

Scope::~Scope() {
Y
Yu Yang 已提交
26
  DropKids();
27 28 29 30
  for (auto& kv : vars_) {
    VLOG(3) << "Destroy variable " << kv.first;
    delete kv.second;
  }
Y
Yi Wang 已提交
31 32
}

Y
Yu Yang 已提交
33
Scope& Scope::NewScope() const {
Y
Yi Wang 已提交
34 35 36 37
  kids_.push_back(new Scope(this));
  return *kids_.back();
}

D
dongzhihong 已提交
38
Variable* Scope::Var(const std::string& name) {
Y
Yi Wang 已提交
39
  auto iter = vars_.find(name);
Y
Yi Wang 已提交
40
  if (iter != vars_.end()) {
Y
Yi Wang 已提交
41
    return iter->second;
Y
Yi Wang 已提交
42 43
  }
  Variable* v = new Variable();
Y
Yi Wang 已提交
44
  vars_[name] = v;
45
  VLOG(3) << "Create variable " << name << " on scope";
Y
Yi Wang 已提交
46
  v->name_ = &(vars_.find(name)->first);
Y
Yi Wang 已提交
47 48 49
  return v;
}

Y
Yu Yang 已提交
50 51 52 53 54 55
Variable* Scope::Var(std::string* name) {
  auto var_name = string::Sprintf("%p.%d", this, vars_.size());
  if (name != nullptr) {
    *name = var_name;
  }
  return Var(var_name);
Y
Yi Wang 已提交
56 57 58 59
}

Variable* Scope::FindVar(const std::string& name) const {
  auto it = vars_.find(name);
Y
Yi Wang 已提交
60
  if (it != vars_.end()) return it->second;
Y
Yi Wang 已提交
61 62 63
  return (parent_ == nullptr) ? nullptr : parent_->FindVar(name);
}

Y
Yu Yang 已提交
64
const Scope* Scope::FindScope(const Variable* var) const {
Y
Yi Wang 已提交
65 66 67 68 69
  for (auto& kv : vars_) {
    if (kv.second == var) {
      return this;
    }
  }
Y
Yi Wang 已提交
70 71
  return (parent_ == nullptr) ? nullptr : parent_->FindScope(var);
}
Y
Yu Yang 已提交
72 73 74 75
void Scope::DropKids() {
  for (Scope* s : kids_) delete s;
  kids_.clear();
}
Y
Yi Wang 已提交
76

77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93
std::vector<std::string> Scope::GetAllNames(bool recursive) const {
  std::vector<std::string> known_vars(vars_.size());

  if (recursive) {
    for (auto& kid : kids_) {
      auto kid_vars = kid->GetAllNames();
      for (auto& p : kid_vars) {
        known_vars.emplace_back(p);
      }
    }
  }
  for (auto& p : vars_) {
    known_vars.emplace_back(p.first);
  }
  return known_vars;
}

Y
Yu Yang 已提交
94 95 96 97 98 99 100
void Scope::DeleteScope(Scope* scope) {
  auto it = std::find(this->kids_.begin(), this->kids_.end(), scope);
  PADDLE_ENFORCE(it != this->kids_.end(), "Cannot find %p as kid scope", scope);
  this->kids_.erase(it);
  delete scope;
}

Y
Yu Yang 已提交
101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118
void Scope::Rename(const std::string& origin_name,
                   const std::string& new_name) const {
  auto origin_it = vars_.find(origin_name);
  PADDLE_ENFORCE(origin_it != vars_.end(),
                 "Cannot find original variable with name %s", origin_name);
  auto new_it = vars_.find(new_name);
  PADDLE_ENFORCE(new_it == vars_.end(),
                 "The variable with name %s is already in the scope", new_name);
  vars_[new_name] = origin_it->second;
  vars_.erase(origin_it);
}

std::string Scope::Rename(const std::string& origin_name) const {
  auto var_name = string::Sprintf("%p.%d", this, vars_.size());
  Rename(origin_name, var_name);
  return var_name;
}

Y
Yi Wang 已提交
119 120
}  // namespace framework
}  // namespace paddle