@@ -3,7 +3,6 @@ PROJECT(RedisAI)
33
44# CMake modules should be included in ${PROJECT_SOURCE_DIR}/opt/cmake/modules
55list (APPEND CMAKE_MODULE_PATH ${PROJECT_SOURCE_DIR} /opt/cmake/modules)
6- list (APPEND CMAKE_MODULE_PATH ${PROJECT_SOURCE_DIR} /opt/GoogleTest)
76
87# Set a default build type if none was specified
98set (default_build_type "Release" )
@@ -50,12 +49,11 @@ set(CMAKE_CXX_FLAGS_DEBUG "${CMAKE_CXX_FLAGS_DEBUG} ${CMAKE_COMMON_FLAGS_DEBUG}"
5049
5150#----------------------------------------------------------------------------------------------
5251
53- option (PACKAGE_UNIT_TESTS "Build unit tests" ON )
5452option (BUILD_TF "Build the TensorFlow backend" ON )
5553option (BUILD_TFLITE "Build the TensorFlow Lite backend" ON )
5654option (BUILD_ORT "Build the ONNXRuntime backend" ON )
5755option (BUILD_TORCH "Build the PyTorch backend" ON )
58-
56+ option (BUILD_REDISAI_LITE "Build the RedisAI Lite Varient" OFF )
5957#----------------------------------------------------------------------------------------------
6058
6159FUNCTION (ADD_LDFLAGS _TARGET NEW_FLAGS)
@@ -114,6 +112,9 @@ ENDIF()
114112
115113SET (CMAKE_C_STANDARD 11)
116114ADD_DEFINITIONS (-DREDISMODULE_EXPERIMENTAL_API)
115+ if (BUILD_REDISAI_LITE)
116+ ADD_DEFINITIONS (-DREDISAI_LITE)
117+ ENDIF ()
117118
118119# SET(CUDA_TOOLKIT_ROOT_DIR /usr/local/cuda-10.0)
119120
@@ -157,6 +158,7 @@ ENDIF()
157158IF (BUILD_ORT)
158159 FIND_LIBRARY (ORT_LIBRARIES NAMES onnxruntime
159160 PATHS ${depsAbs} /onnxruntime/lib)
161+ ADD_SUBDIRECTORY (src/backends/onnx_allocator)
160162 MESSAGE (STATUS "Found ONNXRuntime Libraries: \" ${ORT_LIBRARIES} \" )" )
161163 IF (NOT ORT_LIBRARIES)
162164 MESSAGE (FATAL_ERROR "Could not find ONNXRuntime" )
@@ -290,6 +292,7 @@ ENDIF()
290292
291293IF (BUILD_ORT)
292294 ADD_LIBRARY (redisai_onnxruntime SHARED $<TARGET_OBJECTS:redisai_onnxruntime_obj>)
295+ TARGET_LINK_LIBRARIES (redisai_onnxruntime onnx_allocator ${ORT_LIBRARIES} )
293296 TARGET_LINK_LIBRARIES (redisai_onnxruntime ${ORT_LIBRARIES} )
294297 SET_TARGET_PROPERTIES (redisai_onnxruntime PROPERTIES PREFIX "" )
295298 SET_TARGET_PROPERTIES (redisai_onnxruntime PROPERTIES SUFFIX ".so" )
@@ -316,10 +319,3 @@ ENDIF()
316319IF (NOT ${installAbs} STREQUAL ${CMAKE_SOURCE_DIR} /install -${DEVICE} )
317320 INSTALL_SYMLINK(${installAbs} ${CMAKE_SOURCE_DIR} /install -${DEVICE} )
318321ENDIF ()
319-
320-
321- if (PACKAGE_UNIT_TESTS)
322- enable_testing ()
323- include (GoogleTest)
324- add_subdirectory (tests/unit)
325- endif ()
0 commit comments