update ivf_flat go api

rapidsai · Feb 6, 2025 · 9996ed9 · 9996ed9
1 parent 56ba002
commit 9996ed9
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/go/ivf_flat/ivf_flat.go b/go/ivf_flat/ivf_flat.go
@@ -67,6 +67,10 @@ func SearchIndex[T any](Resources cuvs.Resource, params *SearchParams, index *Iv
 	if !index.trained {
 		return errors.New("index needs to be built before calling search")
 	}
+        prefilter := C.cuvsFilter{
+                addr:  0,
+                _type: C.NO_FILTER,
+        }
 
-	return cuvs.CheckCuvs(cuvs.CuvsError(C.cuvsIvfFlatSearch(C.cuvsResources_t(Resources.Resource), params.params, index.index, (*C.DLManagedTensor)(unsafe.Pointer(queries.C_tensor)), (*C.DLManagedTensor)(unsafe.Pointer(neighbors.C_tensor)), (*C.DLManagedTensor)(unsafe.Pointer(distances.C_tensor)))))
+	return cuvs.CheckCuvs(cuvs.CuvsError(C.cuvsIvfFlatSearch(C.cuvsResources_t(Resources.Resource), params.params, index.index, (*C.DLManagedTensor)(unsafe.Pointer(queries.C_tensor)), (*C.DLManagedTensor)(unsafe.Pointer(neighbors.C_tensor)), (*C.DLManagedTensor)(unsafe.Pointer(distances.C_tensor)), prefilter)))
 }