fix python cmake error

deepmodeling · Aug 16, 2021 · a6aeeab · a6aeeab
1 parent 68577be
commit a6aeeab
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/source/lib/CMakeLists.txt b/source/lib/CMakeLists.txt
@@ -11,7 +11,7 @@ if (USE_CUDA_TOOLKIT)
   add_definitions("-DGOOGLE_CUDA")
   add_subdirectory(src/cuda)
   set (EXTRA_LIBS ${EXTRA_LIBS} deepmd_op_cuda)
-  target_link_libraries (${libname} PRIVATE ${CUDA_LIBRARIES} ${EXTRA_LIBS})
+  target_link_libraries (${libname} INTERFACE ${CUDA_LIBRARIES} ${EXTRA_LIBS})
   # gpu_cuda.h
   target_include_directories(${libname} PUBLIC ${CUDA_INCLUDE_DIRS})
 endif()
@@ -20,7 +20,7 @@ if (USE_ROCM_TOOLKIT)
   add_definitions("-DTENSORFLOW_USE_ROCM")
   add_subdirectory(src/rocm)
   set (EXTRA_LIBS ${EXTRA_LIBS} deepmd_op_rocm)
-  target_link_libraries (${libname} PRIVATE ${ROCM_LIBRARIES} ${EXTRA_LIBS})
+  target_link_libraries (${libname} INTERFACE ${ROCM_LIBRARIES} ${EXTRA_LIBS})
   # gpu_rocm.h
   target_include_directories(${libname} PUBLIC ${ROCM_INCLUDE_DIRS})
 endif()