More fixes

rapidsai · Oct 31, 2023 · b11326c · b11326c
1 parent aa0fada
commit b11326c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/raft-ann-bench/src/raft-ann-bench/validators/__init__.py b/python/raft-ann-bench/src/raft-ann-bench/validators/__init__.py
@@ -27,11 +27,11 @@ def raft_ivf_pq_search_validator(params, build_params, k, batch_size):
     if "internalDistanceDtype" in params and "smemLutDtype" in params:
         ret = (
             DTYPE_SIZES[params["smemLutDtype"]]
-            < DTYPE_SIZES[params["internalDistanceDtype"]]
+            >= DTYPE_SIZES[params["internalDistanceDtype"]]
         )
 
     if "nlist" in build_params and "nprobe" in params:
-        ret = build_params["nlist"] <= params["nprobe"]
+        ret = build_params["nlist"] >= params["nprobe"]
     return ret