From fb81d3fb5c1f994c7291e9b8effa534c8b163239 Mon Sep 17 00:00:00 2001
From: Tamas Bela Feher <tfeher@nvidia.com>
Date: Tue, 31 Oct 2023 02:10:47 +0100
Subject: [PATCH] Add configuration for ANN benchmark tests

---
 .../raft-ann-bench/run/conf/fix_latency.json  | 29 +++++++++++++++++++
 1 file changed, 29 insertions(+)
 create mode 100644 python/raft-ann-bench/src/raft-ann-bench/run/conf/fix_latency.json

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}
+      ]
+    }
+  ]
+}