Skip to content
Projects
Groups
Snippets
Help
Loading...
Sign in / Register
Toggle navigation
N
ngraph
Project
Project
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Packages
Packages
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
submodule
ngraph
Commits
3cf51dc5
Commit
3cf51dc5
authored
Sep 11, 2017
by
Jaikrishnan Menon
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
CMake: Skip MKL-DNN on OSX/Darwin
Currently MKL-DNN isn't supported on this platform
parent
305a9a8a
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
28 additions
and
19 deletions
+28
-19
external_mkldnn.cmake
cmake/external_mkldnn.cmake
+15
-11
CMakeLists.txt
test/CMakeLists.txt
+13
-8
No files found.
cmake/external_mkldnn.cmake
View file @
3cf51dc5
...
@@ -17,12 +17,14 @@ include(ExternalProject)
...
@@ -17,12 +17,14 @@ include(ExternalProject)
# Fetch and install MKL-DNN
# Fetch and install MKL-DNN
#----------------------------------------------------------------------------------------------------------
#----------------------------------------------------------------------------------------------------------
SET
(
MKLDNN_GIT_REPO_URL https://github.com/01org/mkl-dnn
)
if
(
NOT
${
CMAKE_SYSTEM_NAME
}
MATCHES
"Darwin"
)
set
(
EXTERNAL_INSTALL_LOCATION
${
CMAKE_BINARY_DIR
}
/external
)
SET
(
MKLDNN_GIT_REPO_URL https://github.com/01org/mkl-dnn
)
# The 'BUILD_BYPRODUCTS' argument was introduced in CMake 3.2.
set
(
EXTERNAL_INSTALL_LOCATION
${
CMAKE_BINARY_DIR
}
/external
)
if
(
${
CMAKE_VERSION
}
VERSION_LESS 3.2
)
# The 'BUILD_BYPRODUCTS' argument was introduced in CMake 3.2.
if
(
${
CMAKE_VERSION
}
VERSION_LESS 3.2
)
ExternalProject_Add
(
ExternalProject_Add
(
ext_mkldnn
ext_mkldnn
GIT_REPOSITORY
${
MKLDNN_GIT_REPO_URL
}
GIT_REPOSITORY
${
MKLDNN_GIT_REPO_URL
}
...
@@ -30,7 +32,7 @@ if (${CMAKE_VERSION} VERSION_LESS 3.2)
...
@@ -30,7 +32,7 @@ if (${CMAKE_VERSION} VERSION_LESS 3.2)
PATCH_COMMAND git am
${
CMAKE_SOURCE_DIR
}
/third-party/patches/mkldnn-cmake-openmp.patch
PATCH_COMMAND git am
${
CMAKE_SOURCE_DIR
}
/third-party/patches/mkldnn-cmake-openmp.patch
CMAKE_ARGS -DCMAKE_INSTALL_PREFIX=
${
EXTERNAL_INSTALL_LOCATION
}
CMAKE_ARGS -DCMAKE_INSTALL_PREFIX=
${
EXTERNAL_INSTALL_LOCATION
}
)
)
else
()
else
()
ExternalProject_Add
(
ExternalProject_Add
(
ext_mkldnn
ext_mkldnn
GIT_REPOSITORY
${
MKLDNN_GIT_REPO_URL
}
GIT_REPOSITORY
${
MKLDNN_GIT_REPO_URL
}
...
@@ -39,11 +41,11 @@ else()
...
@@ -39,11 +41,11 @@ else()
CMAKE_ARGS -DCMAKE_INSTALL_PREFIX=
${
EXTERNAL_INSTALL_LOCATION
}
CMAKE_ARGS -DCMAKE_INSTALL_PREFIX=
${
EXTERNAL_INSTALL_LOCATION
}
BUILD_BYPRODUCTS
"
${
EXTERNAL_INSTALL_LOCATION
}
/include/mkldnn.hpp"
BUILD_BYPRODUCTS
"
${
EXTERNAL_INSTALL_LOCATION
}
/include/mkldnn.hpp"
)
)
endif
()
endif
()
ExternalProject_Get_Property
(
ext_mkldnn source_dir binary_dir
)
ExternalProject_Get_Property
(
ext_mkldnn source_dir binary_dir
)
ExternalProject_Add_Step
(
ExternalProject_Add_Step
(
ext_mkldnn
ext_mkldnn
PrepareMKL
PrepareMKL
COMMAND
${
source_dir
}
/scripts/prepare_mkl.sh
COMMAND
${
source_dir
}
/scripts/prepare_mkl.sh
...
@@ -51,7 +53,8 @@ ExternalProject_Add_Step(
...
@@ -51,7 +53,8 @@ ExternalProject_Add_Step(
DEPENDERS configure
DEPENDERS configure
)
)
#----------------------------------------------------------------------------------------------------------
set
(
MKLDNN_INCLUDE_DIR
"
${
EXTERNAL_INSTALL_LOCATION
}
/include"
PARENT_SCOPE
)
set
(
MKLDNN_INCLUDE_DIR
"
${
EXTERNAL_INSTALL_LOCATION
}
/include"
PARENT_SCOPE
)
set
(
MKLDNN_LIB_DIR
"
${
EXTERNAL_INSTALL_LOCATION
}
/lib"
PARENT_SCOPE
)
set
(
MKLDNN_LIB_DIR
"
${
EXTERNAL_INSTALL_LOCATION
}
/lib"
PARENT_SCOPE
)
endif
()
\ No newline at end of file
test/CMakeLists.txt
View file @
3cf51dc5
...
@@ -15,23 +15,17 @@ include_directories(
...
@@ -15,23 +15,17 @@ include_directories(
SYSTEM
SYSTEM
${
GTEST_INCLUDE_DIR
}
${
GTEST_INCLUDE_DIR
}
${
EIGEN_INCLUDE_DIR
}
${
EIGEN_INCLUDE_DIR
}
${
MKLDNN_INCLUDE_DIR
}
)
)
include_directories
(
include_directories
(
${
NGRAPH_INCLUDE_DIR
}
${
NGRAPH_INCLUDE_DIR
}
)
)
link_directories
(
${
MKLDNN_LIB_DIR
}
)
set
(
SRC
set
(
SRC
main.cpp
main.cpp
build_graph.cpp
build_graph.cpp
eigen.cpp
eigen.cpp
element_type.cpp
element_type.cpp
mkldnn.cpp
op.cpp
op.cpp
input_output_assign.cpp
input_output_assign.cpp
tensor.cpp
tensor.cpp
...
@@ -41,15 +35,26 @@ set (SRC
...
@@ -41,15 +35,26 @@ set (SRC
uuid.cpp
uuid.cpp
)
)
if
(
MKLDNN_INCLUDE_DIR
)
include_directories
(
${
MKLDNN_INCLUDE_DIR
}
)
link_directories
(
${
MKLDNN_LIB_DIR
}
)
set
(
SRC
${
SRC
}
mkldnn.cpp
)
endif
()
set
(
CMAKE_CXX_FLAGS
"
${
CMAKE_CXX_FLAGS
}
-std=c++11"
)
set
(
CMAKE_CXX_FLAGS
"
${
CMAKE_CXX_FLAGS
}
-std=c++11"
)
set
(
CMAKE_CXX_FLAGS
"
${
CMAKE_CXX_FLAGS
}
-DCURDIR=
\\\"
${
CMAKE_CURRENT_SOURCE_DIR
}
\\\"
"
)
set
(
CMAKE_CXX_FLAGS
"
${
CMAKE_CXX_FLAGS
}
-DCURDIR=
\\\"
${
CMAKE_CURRENT_SOURCE_DIR
}
\\\"
"
)
add_executable
(
unit-test
${
SRC
}
)
add_executable
(
unit-test
${
SRC
}
)
target_link_libraries
(
unit-test ngraph libgtest mkldnn pthread
)
if
(
MKLDNN_INCLUDE_DIR
)
target_link_libraries
(
unit-test mkldnn
)
add_dependencies
(
unit-test ext_mkldnn
)
endif
()
target_link_libraries
(
unit-test ngraph libgtest pthread
)
target_link_libraries
(
unit-test
${
CMAKE_DL_LIBS
}
)
target_link_libraries
(
unit-test
${
CMAKE_DL_LIBS
}
)
add_dependencies
(
unit-test ngraph libgtest eigen
ext_mkldnn
)
add_dependencies
(
unit-test ngraph libgtest eigen
)
add_custom_target
(
check
add_custom_target
(
check
COMMAND
${
PROJECT_BINARY_DIR
}
/test/unit-test
COMMAND
${
PROJECT_BINARY_DIR
}
/test/unit-test
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment