fix mmdeploy_pplnn_net build error when target device is cpu (open-mm…

…lab#896)
RunningLeon · Aug 15, 2022 · 91a060f · 91a060f
1 parent 127125f
commit 91a060f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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})