Function TEST

oneflow/core/embedding/key_value_store_test.cpp:174–192

}

TEST(PersistentTableKeyValueStore, PersistentTableKeyValueStore) {
  if (!HasCudaDevice()) { return; }
  Singleton<ep::DeviceManagerRegistry>::New();
  PersistentTableKeyValueStoreOptions options{};
  uint32_t value_length = 128;

  std::string path = CreateTempDirectory();
  options.table_options.path = path;
  options.table_options.value_size = value_length * sizeof(float);
  options.table_options.key_size = GetSizeOfDataType(DataType::kUInt64);
  options.table_options.physical_block_size = 512;

  std::unique_ptr<KeyValueStore> store = NewPersistentTableKeyValueStore(options);
  store->ReserveQueryLength(128);
  TestKeyValueStore(store.get(), 1024, 1024, value_length);
  store.reset();
  PosixFile::RecursiveDelete(path);
  Singleton<ep::DeviceManagerRegistry>::Delete();
}

TEST(CachedKeyValueStore, LRU) {
  if (!HasCudaDevice()) { return; }