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

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
@yajiedesign
yajiedesign committed Apr 20, 2018
1 parent 48f26a3 commit c5053fc
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions CMakeLists.txt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -162,6 +162,9 @@ if(USE_MKLDNN)
add_subdirectory(3rdparty/mkldnn)
include_directories(3rdparty/mkldnn/include)
add_definitions(-DUSE_MKL=1)
add_definitions(-DCUB_MKL=1)
add_definitions(-DMXNET_USE_MKLDNN=1)


if(MSVC)
target_compile_definitions(mkldnn PRIVATE WIN32)
Expand Down

0 comments on commit c5053fc

Please sign in to comment.