diff options
| author |  Eugene Zhulenev <ezhulenev@google.com> | 2018-09-26 11:08:47 -0700 |
|---|---|---|
| committer | Eugene Zhulenev <ezhulenev@google.com> | 2018-09-26 11:08:47 -0700 |
| commit | 71cd3fbd6a03991977fd3faf82109bf27b701d2d (patch) | |
| tree | bb017d89f9901d9d6d550eb20f9c5af3d0dd2aae /cmake | |
| parent | 0a3356f4ece30cd486b616eb1da9128aa4f245be (diff) | |
Support multiple contraction kernel types in TensorContractionThreadPool
Diffstat (limited to '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 |