Class InterpreterCoreInfoCache

paddle/fluid/framework/executor_cache.h:142–180

};

class InterpreterCoreInfoCache {
 public:
  static InterpreterCoreInfoCache& Instance();
  bool Has(const InterpreterCoreInfoCacheKey& key) const {
    int64_t hash_key = key.hash();
    return info_map_.find(hash_key) != info_map_.end() &&
           info_map_.at(hash_key).IsAvailable(key.is_grad());
  }

  InterpreterCoreInfo::CacheValue& GetMutable(
      const InterpreterCoreInfoCacheKey& key) {
    int64_t hash_key = key.hash();
    return info_map_[hash_key].GetMutable(key.is_grad());
  }

  void UpdateSkipEagerDeleteVars(const InterpreterCoreInfoCacheKey& key,
                                 const std::set<std::string>& skip_vars) {
    auto& cached_value = GetMutable(key);
    cached_value.skip_eager_delete_vars_ = std::move(skip_vars);
  }

  std::set<std::string>& GetSkipEagerDeleteVars(
      const InterpreterCoreInfoCacheKey& key) {
    auto& cached_value = GetMutable(key);
    return cached_value.skip_eager_delete_vars_;
  }

  size_t Size() const { return info_map_.size(); }

  void Finalize() {
    // NOTE(Aurelius84): DO NOT perform finalize in destructor
    // to avoid problems caused by destructor order of static
    // object.
    info_map_.clear();
  }

 private:
  std::unordered_map<int64_t, InterpreterCoreInfo> info_map_;
};

std::shared_ptr<InterpreterCore> CreateProgramInterpreterCoreInfoToCache(
    const ProgramDesc& program_desc,