add dlpack to imdb demo, test=develop (#21069)

6 years ago · b592deec90
parent 7ee25189c3
commit b592deec90
1 changed files with 1 additions and 0 deletions
--- a/paddle/fluid/train/imdb_demo/CMakeLists.txt
+++ b/paddle/fluid/train/imdb_demo/CMakeLists.txt
@ -20,6 +20,7 @@ include_directories("${PADDLE_LIB}/third_party/install/zlib/include")

 include_directories("${PADDLE_LIB}/third_party/boost")
 include_directories("${PADDLE_LIB}/third_party/eigen3")
+include_directories("${PADDLE_LIB}/third_party/dlpack")

 link_directories("${PADDLE_LIB}/third_party/install/protobuf/lib")
 link_directories("${PADDLE_LIB}/third_party/install/glog/lib")