Function SingleCommit

tensorflow/core/kernels/embedding_variable_ops_test.cc:1161–1178

}

void SingleCommit(KVInterface<int64, float>* hashmap,
    std::vector<int64> keys, int bias) {
  std::vector<void*> value_ptrs;
  for (int64 i = 0; i < keys.size(); ++i) {
    void* tmp = cpu_allocator()->AllocateRaw(0, 124 * sizeof(float) + 16);
    for (int j = 0; j < 124; j++) {
      ((float*)tmp)[j] = keys[i] + bias;
    }
    value_ptrs.push_back(tmp);
  }
  ASSERT_EQ(keys.size(), value_ptrs.size());
  uint64 start = Env::Default()->NowNanos();
  for (int64 i = 0; i < keys.size(); i++) {
    hashmap->Commit(keys[i], value_ptrs[i]);
  }
  uint64 end = Env::Default()->NowNanos();
  uint64 result_cost = end - start;
}

void TestCompaction() {
  std::string temp_dir = testing::TmpDir();