EXAMPLES/UCS_ANN_SEARCH: new example to be also used for benchmarking

Author: alex--m

examples/Makefile.am

@@ -6,11 +6,12 @@
 #
 
 examplesdir = $(pkgdatadir)/examples
-dist_examples_DATA =   \
+dist_examples_DATA = \
 	hello_world_util.h \
 	ucp_hello_world.c  \
 	uct_hello_world.c  \
 	ucp_client_server.c \
+	ucs_ann_search.c \
 	ucp_util.h
 
 if HAVE_CUDA
@@ -43,6 +44,13 @@ noinst_PROGRAMS = \
 	uct_hello_world \
 	ucp_client_server
 
+lib_LTLIBRARIES = libucs_ann_search.la
+
+libucs_ann_search_la_SOURCES  = ucs_ann_search.c
+libucs_ann_search_la_LIBADD   = $(top_builddir)/src/ucs/libucs.la
+libucs_ann_search_la_LDFLAGS  = -ldl -version-info $(SOVERSION)
+libucs_ann_search_la_CPPFLAGS = $(BASE_CPPFLAGS) -g
+
 ucp_hello_world_SOURCES  = ucp_hello_world.c
 ucp_hello_world_CFLAGS   = $(BASE_CFLAGS) $(EXAMPLE_CUDA_CFLAGS)
 ucp_hello_world_CPPFLAGS = $(BASE_CPPFLAGS) $(EXAMPLE_CUDA_CPPFLAGS)

examples/ucs_ann_search.c

@@ -0,0 +1,31 @@
+/**
+ * See file LICENSE for terms.
+ */
+
+#include <ucs/datastruct/ndim_knn.h>
+
+#define EXPORTED __attribute__((visibility("protected")))
+#define DIM_LIST 1, 2, 3, 16, 20, 25, 50, 96, 100, 128, 200, 256, 784, 800, 960
+
+#define WNN_BY_DIM(_K, _ndim) \
+    NDIM_WNN_INIT2(char_to_unsigned_##_ndim##_dim_k##_K, _K, _ndim, EXPORTED, \
+                   char, unsigned, khint32_t, kh_int_hash_func, \
+                   kh_int_hash_equal, kh_int_hash_func, kh_int_hash_equal) \
+    NDIM_WNN_INIT2(int32_to_int32_##_ndim##_dim_k##_K, _K, _ndim, EXPORTED, \
+                   khint32_t, unsigned, khint32_t, kh_int_hash_func, \
+                   kh_int_hash_equal, kh_int_hash_func, kh_int_hash_equal) \
+    NDIM_WNN_INIT2(int64_to_int64_##_ndim##_dim_k##_K, _K, _ndim, EXPORTED, \
+                   khint64_t, khint64_t, khint64_t, kh_int_hash_func, \
+                   kh_int_hash_equal, kh_int64_hash_func, kh_int64_hash_equal) \
+    NDIM_WNN_INIT2(float_to_unsigned_##_ndim##_dim_k##_K, _K, _ndim, EXPORTED, \
+                   float, unsigned, int, kh_int_hash_func, kh_int_hash_equal, \
+                   kh_int_hash_func, kh_int_hash_equal) \
+    NDIM_WNN_INIT2(float_to_ptr_##_ndim##_dim_k##_K, _K, _ndim, EXPORTED, \
+                   float, uintptr_t, float, kh_int_hash_func, kh_int_hash_equal, \
+                   kh_int64_hash_func, kh_int64_hash_equal) \
+    NDIM_WNN_INIT2(cstr_to_ptr_##_ndim##_dim_k##_K, _K, _ndim, EXPORTED, \
+                   kh_cstr_t, unsigned, float, kh_str_hash_func, \
+                   kh_str_hash_equal, kh_int_hash_func, kh_int_hash_equal)
+
+UCS_PP_FOREACH(WNN_BY_DIM,  1, DIM_LIST)
+UCS_PP_FOREACH(WNN_BY_DIM, 10, DIM_LIST)