diff --git a/vector.go b/vector.go index 7a873e7..282413b 100644 --- a/vector.go +++ b/vector.go @@ -95,3 +95,15 @@ const TrainingKey = "_training" const TrainedIndexCallback = "_trained_index_callback" type TrainedIndexCallbackFn func(string) (interface{}, error) + +// GPU vector search error callbacks + +// Called when an index move from CPU to CPU fails +const CPUToGPUCloneErrorKey = "_cpu_to_gpu_clone_error" + +type CPUToGPUCloneErrorCallback func(err error) + +// Called when a generic error occurs on the GPU +const GPUErrorKey = "_gpu_error" + +type GPUErrorCallback func(err error)