Function PerfMemory

tensorflow/core/kernels/embedding_variable_memory_test.cc:19–43

namespace tensorflow {
namespace embedding {
float PerfMemory(Tensor& default_value,
                const std::vector<int64>& id_list,
                int value_size, int64 default_value_dim,
                int64 filter_freq = 0, int64 steps_to_live = 0,
                int64 record_freq = false) {
  auto ev = CreateEmbeddingVar(value_size, default_value,
                               default_value_dim, filter_freq,
                               steps_to_live, -1.0,
                               embedding::StorageType::DRAM,
                               {1024, 1024, 1024, 1024},
                               record_freq);
  void* value_ptr = nullptr;
  bool is_filter = false;
  double start_mem, end_mem;
  start_mem = getResident() * getpagesize();
  for (int i = 0; i < id_list.size(); i++) {
    ev->LookupOrCreateKey(id_list[i], &value_ptr, &is_filter, false);
    if (is_filter)
      ev->flat(value_ptr);
  }
  end_mem = getResident() * getpagesize();
  double used_mb = (end_mem - start_mem)/1000000;
  LOG(INFO)<<"[TestMemory]Use Memory: "<<used_mb;
  return used_mb;
}

TEST(EmbeddingVariabelMemoryTest, TestMemory) {
  int value_size = 32;