Function TEST

tensorflow/core/kernels/embedding_variable_performance_test.cc:183–206

}

TEST(EmbeddingVariablePerformanceTest, TestLookupOrCreate) {
  int num_of_batch = 100;
  int batch_size = 1024 * 128;
  int num_of_ids = 5000000;
  std::vector<std::vector<int64>> input_batches(num_of_batch);
  for (int i = 0; i < num_of_batch; i++) {
    input_batches[i].resize(batch_size);
  }
  LOG(INFO)<<"[TestLookupOrCreate] Start generating skew input";
  GenerateSkewInput(num_of_ids, 0.8, input_batches);
  LOG(INFO)<<"[TestLookupOrCreate] Finish generating skew input";
  std::vector<int> num_thread_vec({1, 2, 4, 8, 16});
  for (auto num_thread: num_thread_vec) {
    LOG(INFO)<<"[TestLookupOrCreate] Test LookupOrCreate With "
             <<num_thread<<" threads.";
    double exec_time = PerfLookupOrCreate(input_batches, num_thread);
    if (exec_time == -1.0) {
      LOG(INFO)<<"[TestLookupOrCreate] Test Failed";
    } else {
      LOG(INFO)<<"[TestLookupOrCreate] Performance of LookupOrCreate With "
               <<num_thread<<" threads: "<<exec_time/1000000<<" ms";
    }
  }
}

void thread_lookup(
    EmbeddingVar<int64, float>* ev,