add the missing macro PADDLE_USE_MKLDNN

tensor-tang · tensor-tang · commit 1578c20aaf47 · 2017-11-17T13:28:56.000+08:00
diff --git a/cmake/external/mkldnn.cmake b/cmake/external/mkldnn.cmake
@@ -67,4 +67,5 @@ ADD_LIBRARY(mkldnn SHARED IMPORTED GLOBAL)
 SET_PROPERTY(TARGET mkldnn PROPERTY IMPORTED_LOCATION ${MKLDNN_LIB})
 ADD_DEPENDENCIES(mkldnn ${MKLDNN_PROJECT})
 MESSAGE(STATUS "MKLDNN library: ${MKLDNN_LIB}")
+add_definitions(-DPADDLE_USE_MKLDNN)
 LIST(APPEND external_project_dependencies mkldnn)
