diff --git a/python/raft-ann-bench/src/raft-ann-bench/run/conf/fix_latency.json b/python/raft-ann-bench/src/raft-ann-bench/run/conf/fix_latency.json new file mode 100644 index 0000000000..54337af832 --- /dev/null +++ b/python/raft-ann-bench/src/raft-ann-bench/run/conf/fix_latency.json @@ -0,0 +1,29 @@ +{ + "dataset": { + "name": "dummy", + "base_file": "deep-1B/base.1B.fbin", + "subset_size": 10, + "query_file": "deep-1B/query.public.10K.fbin", + "groundtruth_neighbors_file": "deep-1B/groundtruth.neighbors.ibin", + "distance": "euclidean" + }, + + "search_basic_param": { + "batch_size": 10000, + "k": 10 + }, + + "index": [ + { + "name": "fix_latency", + "algo": "fix_latency", + "build_param": {"use_gpu": true}, + "file": "fix_latency/dummy_idx", + "search_params": [ + {"use_gpu": false}, + {"use_gpu": true, "sync_stream": false}, + {"use_gpu": true, "sync_stream": true} + ] + } + ] +}