diff --git a/apps/search_memory_index.cpp b/apps/search_memory_index.cpp index 4af0266d6..ca3045331 100644 --- a/apps/search_memory_index.cpp +++ b/apps/search_memory_index.cpp @@ -247,8 +247,14 @@ int search_memory_index(diskann::Metric &metric, const std::string &index_path, diskann::cout << "Ignoring search with L:" << L << " since it's smaller than K:" << recall_at << std::endl; continue; } - std::string cur_result_path = result_path_prefix + "_" + std::to_string(L) + "_idx_uint32.bin"; + std::string cur_result_path_prefix = result_path_prefix + "_" + std::to_string(L); + + std::string cur_result_path = cur_result_path_prefix + "_idx_uint32.bin"; diskann::save_bin(cur_result_path, query_result_ids[test_id].data(), query_num, recall_at); + + cur_result_path = cur_result_path_prefix + "_dists_float.bin"; + diskann::save_bin(cur_result_path, query_result_dists[test_id].data(), query_num, recall_at); + test_id++; }