diff --git a/plugins/CMakeLists.txt b/plugins/CMakeLists.txt index a2e7c77..f42537e 100644 --- a/plugins/CMakeLists.txt +++ b/plugins/CMakeLists.txt @@ -209,7 +209,10 @@ set_target_properties(preprocess PROPERTIES ) # ai_yolo plugin (RKNN-based YOLO inference) -add_library(ai_yolo SHARED ai_yolo/ai_yolo_node.cpp) +add_library(ai_yolo SHARED + ai_yolo/ai_yolo_node.cpp + ${CMAKE_SOURCE_DIR}/src/utils/dma_alloc.cpp +) target_include_directories(ai_yolo PRIVATE ${CMAKE_SOURCE_DIR}/include ${CMAKE_SOURCE_DIR}/third_party) target_link_libraries(ai_yolo PRIVATE project_options Threads::Threads ai_scheduler) if(RK3588_ENABLE_RKNN AND RK_RKNN_LIB) @@ -224,7 +227,10 @@ set_target_properties(ai_yolo PROPERTIES ) # ai_face_det plugin (RKNN-based RetinaFace inference) -add_library(ai_face_det SHARED ai_face_det/ai_face_det_node.cpp) +add_library(ai_face_det SHARED + ai_face_det/ai_face_det_node.cpp + ${CMAKE_SOURCE_DIR}/src/utils/dma_alloc.cpp +) target_include_directories(ai_face_det PRIVATE ${CMAKE_SOURCE_DIR}/include ${CMAKE_SOURCE_DIR}/third_party) target_link_libraries(ai_face_det PRIVATE project_options Threads::Threads ai_scheduler) if(RK3588_ENABLE_RKNN AND RK_RKNN_LIB) @@ -239,7 +245,10 @@ set_target_properties(ai_face_det PROPERTIES ) # ai_face_recog plugin (RKNN-based ArcFace/MobileFaceNet inference) -add_library(ai_face_recog SHARED ai_face_recog/ai_face_recog_node.cpp) +add_library(ai_face_recog SHARED + ai_face_recog/ai_face_recog_node.cpp + ${CMAKE_SOURCE_DIR}/src/utils/dma_alloc.cpp +) target_include_directories(ai_face_recog PRIVATE ${CMAKE_SOURCE_DIR}/include ${CMAKE_SOURCE_DIR}/third_party) target_link_libraries(ai_face_recog PRIVATE project_options Threads::Threads ai_scheduler) if(SQLite3_FOUND)