Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

cmake: copy dnnl headers to include/mkldnn #17647

Merged
merged 3 commits into from
Feb 25, 2020
Merged

cmake: copy dnnl headers to include/mkldnn #17647

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
f0cf878
copy dnnl headers to include/mkldnn
TaoLv Feb 21, 2020
2da48df
trigger ci
TaoLv Feb 21, 2020
3cc8632
Merge branch 'master' of https://github.com/apache/incubator-mxnet in…
TaoLv Feb 22, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 8 additions & 0 deletions CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -756,6 +756,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)
Expand Down