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

[MXNET-343]fix Mkldnn with msvc #10629

Merged
merged 17 commits into from
May 7, 2018
Merged
Changes from 1 commit
Commits
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
Prev Previous commit
Next Next commit
close "-mtune=generic" with msvc, turn off test example
  • Loading branch information
yajiedesign committed May 4, 2018
commit 4188015231ae6efe33b7270a9d6b94e99c3f34c2
7 changes: 6 additions & 1 deletion CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -185,9 +185,14 @@ if(USE_VTUNE)
endif()

if(USE_MKLDNN)
# CPU architecture (e.g., C5) can't run on another architecture (e.g., g3).
# CPU architecture (e.g., C5) can't run on another architecture (e.g., g3).
if(NOT MSVC)
set(ARCH_OPT_FLAGS "-mtune=generic")
endif()
set(WITH_TEST OFF)
set(WITH_EXAMPLE OFF)
add_subdirectory(3rdparty/mkldnn)

include_directories(3rdparty/mkldnn/include)
add_definitions(-DUSE_MKL=1)
add_definitions(-DCUB_MKL=1)
Expand Down