@@ -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" )
@@ -23,7 +22,7 @@ ENDIF()
2322
2423#----------------------------------------------------------------------------------------------
2524
26- SET (CMAKE_CC_COMMON_FLAGS "-fPIC" )
25+ SET (CMAKE_CC_COMMON_FLAGS "-fPIC -fcommon " )
2726IF (USE_PROFILE)
2827 SET (CMAKE_CC_COMMON_FLAGS "${CMAKE_CC_COMMON_FLAGS} -g -ggdb -fno-omit-frame-pointer" )
2928ENDIF ()
@@ -50,7 +49,6 @@ 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 )
@@ -160,6 +158,7 @@ ENDIF()
160158IF (BUILD_ORT)
161159 FIND_LIBRARY (ORT_LIBRARIES NAMES onnxruntime
162160 PATHS ${depsAbs} /onnxruntime/lib )
161+ ADD_SUBDIRECTORY (src/backends/onnx_allocator )
163162 MESSAGE (STATUS "Found ONNXRuntime Libraries: \" ${ORT_LIBRARIES} \" )" )
164163 IF (NOT ORT_LIBRARIES)
165164 MESSAGE (FATAL_ERROR "Could not find ONNXRuntime" )
@@ -293,6 +292,7 @@ ENDIF()
293292
294293IF (BUILD_ORT)
295294 ADD_LIBRARY (redisai_onnxruntime SHARED $<TARGET_OBJECTS :redisai_onnxruntime_obj >)
295+ TARGET_LINK_LIBRARIES (redisai_onnxruntime onnx_allocator ${ORT_LIBRARIES} )
296296 TARGET_LINK_LIBRARIES (redisai_onnxruntime ${ORT_LIBRARIES} )
297297 SET_TARGET_PROPERTIES (redisai_onnxruntime PROPERTIES PREFIX "" )
298298 SET_TARGET_PROPERTIES (redisai_onnxruntime PROPERTIES SUFFIX ".so" )
@@ -319,10 +319,3 @@ ENDIF()
319319IF (NOT ${installAbs} STREQUAL ${CMAKE_SOURCE_DIR} /install-${DEVICE} )
320320 INSTALL_SYMLINK (${installAbs} ${CMAKE_SOURCE_DIR} /install-${DEVICE} )
321321ENDIF ()
322-
323-
324- if (PACKAGE_UNIT_TESTS)
325- enable_testing ()
326- include (GoogleTest )
327- add_subdirectory (tests/unit )
328- endif ()
0 commit comments