fix undefined reference

Signed-off-by: Jinzhe Zeng <[email protected]>
njzjz · Sep 30, 2023 · 6811b1c · 6811b1c
1 parent a922b7a
commit 6811b1c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/source/lib/CMakeLists.txt b/source/lib/CMakeLists.txt
@@ -13,7 +13,8 @@ if(USE_CUDA_TOOLKIT)
   add_definitions("-DGOOGLE_CUDA")
   add_subdirectory(src/gpu)
   set(EXTRA_LIBS ${EXTRA_LIBS} deepmd_op_cuda)
-  target_link_libraries(${libname} INTERFACE deepmd_dyn_cudart ${EXTRA_LIBS})
+  target_link_libraries(${libname} PUBLIC deepmd_dyn_cudart)
+  target_link_libraries(${libname} INTERFACE ${EXTRA_LIBS})
   # gpu_cuda.h
   target_include_directories(
     ${libname} PUBLIC $<BUILD_INTERFACE:${CUDAToolkit_INCLUDE_DIRS}>