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

CMake ignores USE_MKLDNN flag for 1.2.0.RC[0-2] releases #10801

Closed
anirudh2290 opened this issue May 3, 2018 · 1 comment
Closed

CMake ignores USE_MKLDNN flag for 1.2.0.RC[0-2] releases #10801

anirudh2290 opened this issue May 3, 2018 · 1 comment

Comments

@anirudh2290
Copy link
Member

CMake ignores the USE_MKLDNN flag for 1.2.RC2 release. This has been fixed in the master as part of #10731. To workaround the issue on 1.2.RC2 release, please apply the following changes in CMakeLists.txt:

 if(USE_MKL_IF_AVAILABLE)
   if(USE_MKLDNN)
+    set(ARCH_OPT_FLAGS "-mtune=generic")
     add_subdirectory(3rdparty/mkldnn)
     include_directories(3rdparty/mkldnn/include)
+    add_definitions(-DMXNET_USE_MKLDNN=1)
     list(APPEND mxnet_LINKER_LIBS mkldnn)
   endif()
   find_package(MKL)
@roywei
Copy link
Member

roywei commented May 4, 2018

@anirudh2290 could you add relevant labels (installation/build/MKL) to keep track? Thanks

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Projects
None yet
Development

No branches or pull requests

2 participants