diff --git a/csrc/mmdeploy/net/ppl/CMakeLists.txt b/csrc/mmdeploy/net/ppl/CMakeLists.txt index 91f086468b..a65c64d4de 100644 --- a/csrc/mmdeploy/net/ppl/CMakeLists.txt +++ b/csrc/mmdeploy/net/ppl/CMakeLists.txt @@ -14,7 +14,8 @@ if (PPLNN_USE_CUDA AND ("cuda" IN_LIST MMDEPLOY_TARGET_DEVICES)) target_compile_definitions(${PROJECT_NAME} PRIVATE -DPPL_NN_HAS_CUDA=1) target_include_directories(${PROJECT_NAME} PUBLIC ${CUDA_TOOLKIT_ROOT_DIR}/include) target_link_directories(${PROJECT_NAME} PUBLIC ${CUDA_TOOLKIT_ROOT_DIR}/lib64) + target_link_libraries(${PROJECT_NAME} PRIVATE nvrtc) endif () target_link_libraries(${PROJECT_NAME} - PRIVATE ${PPLNN_LIBRARIES} nvrtc) + PRIVATE ${PPLNN_LIBRARIES}) add_library(mmdeploy::pplnn_net ALIAS ${PROJECT_NAME})