Skip to content

Commit 2bb7fcd

Browse files
Revert benchmark to use deep/seismic datasets
1 parent 19bf5f7 commit 2bb7fcd

1 file changed

Lines changed: 3 additions & 5 deletions

File tree

benchmark/bm_Sofa_L2Square.cpp

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ struct SofaSearchOnlyFixture : public benchmark::Fixture {
2727

2828
void SetUp(const benchmark::State& state) override {
2929
int config_idx = static_cast<int>(state.range(0));
30-
const SSTestConfig& config = test_configs[config_idx];
30+
const SSTestConfig& config = test_configs_deep_seismic_astro270m[config_idx];
3131

3232
const bool use_fvecs = endsWith(config.dataset_path, ".fvecs") || endsWith(config.query_path, ".fvecs");
3333
size_t dim_u = 0, n_database_u = 0, n_q_u = 0;
@@ -146,11 +146,9 @@ BENCHMARK_DEFINE_F(SofaSearchOnlyFixture, BM_Sofa_SearchOnly)(benchmark::State&
146146
}
147147
}
148148

149-
// test_configs: Astronomy + RandomWalk, threads={1,4,8}, k={1,10,100} => 18 configs (indices 0-17)
150149
BENCHMARK_REGISTER_F(SofaSearchOnlyFixture, BM_Sofa_SearchOnly)
151-
->Args({0})->Args({1})->Args({2})->Args({3})->Args({4})->Args({5})
152-
->Args({6})->Args({7})->Args({8})->Args({9})->Args({10})->Args({11})
153-
->Args({12})->Args({13})->Args({14})->Args({15})->Args({16})->Args({17})
150+
// q=100, k=1,10,100,1000: DEEP (0-3), Seismic (4-7)
151+
->Args({0})->Args({1})->Args({2})->Args({3})->Args({4})->Args({5})->Args({6})->Args({7})
154152
->Iterations(1)
155153
->Unit(benchmark::kMillisecond);
156154

0 commit comments

Comments
 (0)