MCPcopy Create free account
hub / github.com/alibaba/proxima / LoadFromVecsFile

Function LoadFromVecsFile

tools/bench_client.cc:199–223  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

197}
198
199static bool LoadFromVecsFile() {
200 tools::VecsReader reader;
201 if (!reader.load(FLAGS_file)) {
202 LOG_ERROR("Load vecs file failed.");
203 return false;
204 }
205
206 for (uint32_t i = 0; i < reader.num_vecs(); i++) {
207 Record new_record;
208 uint64_t key = reader.get_key(i);
209 const char *feature = (const char *)reader.get_vector(i);
210
211 new_record.key = key;
212 new_record.vector.append(feature, reader.index_meta().element_size());
213 new_record.dimension = reader.index_meta().dimension();
214 g_record_list.emplace_back(new_record);
215
216 // Limit rows count
217 if (FLAGS_rows > 0 && g_record_list.size() >= FLAGS_rows) {
218 break;
219 }
220 }
221
222 return true;
223}
224
225static bool LoadRecords() {
226 bool ret;

Callers 1

LoadRecordsFunction · 0.70

Calls 8

num_vecsMethod · 0.80
get_keyMethod · 0.80
get_vectorMethod · 0.80
loadMethod · 0.45
appendMethod · 0.45
element_sizeMethod · 0.45
dimensionMethod · 0.45
sizeMethod · 0.45

Tested by

no test coverage detected