add the missing macro PADDLE_USE_MKLDNN

1578c20a · tensor-tang · cece368d · 1578c20a
显示空白变更内容
内联并排

Showing with 1 addition and 0 deletion

cmake/external/mkldnn.cmake cmake/external/mkldnn.cmake +1 -0

未找到文件。
--- 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)