support DCU(rocm)

deepmodeling · Sep 29, 2024 · f1cae59 · f1cae59
1 parent 49ba5a5
commit f1cae59
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/source/CMakeLists.txt b/source/CMakeLists.txt
@@ -51,6 +51,7 @@ if(ENABLE_PADDLE)
   link_directories("${PADDLE_LIB_THIRD_PARTY_PATH}gflags/lib")
   link_directories("${PADDLE_LIB_THIRD_PARTY_PATH}xxhash/lib")
   link_directories("${PADDLE_INFERENCE_DIR}/paddle/lib")
+  add_definitions(-D_GLIBCXX_USE_CXX11_ABI=1)
 endif(ENABLE_PADDLE)
 
 if(BUILD_TESTING)

diff --git a/source/api_cc/CMakeLists.txt b/source/api_cc/CMakeLists.txt
@@ -26,6 +26,10 @@ endif()
 if(ENABLE_PADDLE AND NOT BUILD_PY_IF)
   target_link_libraries(${libname} PUBLIC "${PADDLE_LIBRARIES}")
   target_compile_definitions(${libname} PUBLIC BUILD_PADDLE)
+  if(DP_VARIANT STREQUAL "rocm")
+    target_link_libraries(${libname}
+                          PUBLIC "${hip_LIB_INSTALL_DIR}/libgalaxyhip.so")
+  endif()
 endif()
 
 target_include_directories(