CMake refine for HIP support.

Fix CI.
7 years ago · e0ac6bc436
parent e50205e744
commit e0ac6bc436
1 changed files with 2 additions and 2 deletions
--- a/paddle/fluid/pybind/CMakeLists.txt
+++ b/paddle/fluid/pybind/CMakeLists.txt
@ -1,12 +1,12 @@
 if(WITH_PYTHON)
  if(WITH_AMD_GPU)
    hip_library(paddle_pybind SHARED
-      SRCS pybind.cc exception.cc protobuf.cc const_value.cc
+      SRCS pybind.cc exception.cc protobuf.cc const_value.cc recordio.cc
      DEPS pybind python backward proto_desc paddle_memory executor prune init profiler feed_fetch_method
      ${GLOB_OP_LIB})
  else()
    cc_library(paddle_pybind SHARED
-      SRCS pybind.cc exception.cc protobuf.cc const_value.cc
+      SRCS pybind.cc exception.cc protobuf.cc const_value.cc recordio.cc
      DEPS pybind python backward proto_desc paddle_memory executor prune init profiler feed_fetch_method
      ${GLOB_OP_LIB})
    if(NOT APPLE AND NOT ANDROID)