From f0cf87816756f0a8f1e466ceb2adcc4d6cf2c0f2 Mon Sep 17 00:00:00 2001 From: Tao Lv Date: Fri, 21 Feb 2020 11:38:54 +0800 Subject: [PATCH 1/2] copy dnnl headers to include/mkldnn --- CMakeLists.txt | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/CMakeLists.txt b/CMakeLists.txt index f59c7b999e17..f1b3dd4c120c 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -751,6 +751,14 @@ if(USE_DIST_KVSTORE) list(APPEND mxnet_LINKER_LIBS pslite) endif() +if(USE_MKLDNN) + add_custom_command(TARGET mxnet POST_BUILD + COMMAND ${CMAKE_COMMAND} -E copy + ${CMAKE_BINARY_DIR}/3rdparty/mkldnn/include/dnnl_config.h ${CMAKE_SOURCE_DIR}/include/mkldnn/ + COMMAND ${CMAKE_COMMAND} -E copy + ${CMAKE_BINARY_DIR}/3rdparty/mkldnn/include/dnnl_version.h ${CMAKE_SOURCE_DIR}/include/mkldnn/) +endif() + function(BuildTVMOP) # scope the variables in BuildTVM.cmake to avoid conflict include(cmake/BuildTVM.cmake) From 2da48df6538dd96e0cd8c965250cbc99ac2ffb6a Mon Sep 17 00:00:00 2001 From: Tao Lv Date: Sat, 22 Feb 2020 00:01:57 +0800 Subject: [PATCH 2/2] trigger ci