diff options
Diffstat (limited to 'cmake/FindMkldnn.cmake')
-rw-r--r-- | cmake/FindMkldnn.cmake | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/cmake/FindMkldnn.cmake b/cmake/FindMkldnn.cmake new file mode 100644 index 000000000..6acba9d8c --- /dev/null +++ b/cmake/FindMkldnn.cmake @@ -0,0 +1,18 @@ +# Intel mkl-dnn support. +# Link: https://github.com/intel/mkl-dnn +if (MKLDNN) + set(MKLDNN_FIND_QUIETLY TRUE) + set(MKLDNN_INCLUDES ${MKLDNN}/include) + set(MKLDNN_LIBRARIES ${MKLDNN}/lib) +endif (MKLDNN) +find_path(MKLDNN + NAMES + mkldnn.h + PATHS + $ENV{MKLDNNDIR}/include + ${INCLUDE_INSTALL_DIR} + ) +include(FindPackageHandleStandardArgs) +find_package_handle_standard_args(MKLDNN DEFAULT_MSG + MKLDNN) +mark_as_advanced(MKLDNN)
\ No newline at end of file |