add bitwisehamming distance

rapidsai · Feb 5, 2025 · ef9acc8 · ef9acc8
1 parent 1c1a9d0
commit ef9acc8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/python/cuvs/cuvs/distance/distance.pyx b/python/cuvs/cuvs/distance/distance.pyx
@@ -46,6 +46,7 @@ DISTANCE_TYPES = {
     "minkowski": cuvsDistanceType.LpUnexpanded,
     "russellrao": cuvsDistanceType.RusselRaoExpanded,
     "dice": cuvsDistanceType.DiceExpanded,
+    "bitwise_hamming": cuvsDistanceType.BitwiseHamming
 }
 
 DISTANCE_NAMES = {v: k for k, v in DISTANCE_TYPES.items()}

diff --git a/python/cuvs/cuvs/distance_type.pxd b/python/cuvs/cuvs/distance_type.pxd
@@ -38,4 +38,5 @@ cdef extern from "cuvs/distance/distance.h" nogil:
         KLDivergence
         RusselRaoExpanded
         DiceExpanded
+        BitwiseHamming
         Precomputed