OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Fix IDEEP CMakefile (#9217)

Browse Source 
Summary:
The reason is that we are referencing `__ideep_looked_for` here: 77484d91db/cmake/Modules/FindMKL.cmake (L350)

This was first flushed out in https://github.com/pytorch/pytorch/pull/8105 and probably can help with https://github.com/pytorch/pytorch/issues/9024
Pull Request resolved: https://github.com/pytorch/pytorch/pull/9217

Reviewed By: houseroad

Differential Revision: D8754491

Pulled By: yinghai

fbshipit-source-id: 70aecc2d60684b9ea522403dc98a0a1a2c3db7e6
This commit is contained in:
Yinghai Lu 2018-07-06 15:23:58 -07:00 committed by Facebook Github Bot
parent af107c4d16
commit 766fa1fc96
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
cmake/Modules/FindMKL.cmake
View File

@ -304,7 +304,7 @@ if (USE_MKL AND USE_IDEEP)
if (MKLDNN_INCLUDE_DIR)
list(APPEND IDEEP_INCLUDE_DIR ${MKLDNN_INCLUDE_DIR})
list(APPEND __ideep_looked_for ${MKLDNN_INCLUDE_DIR})
list(APPEND __ideep_looked_for MKLDNN_INCLUDE_DIR)
# to avoid adding conflicting submodels
set(ORIG_WITH_TEST ${WITH_TEST})
set(WITH_TEST OFF)
@ -325,7 +325,7 @@ if (USE_MKL AND USE_IDEEP)
endif()
get_filename_component(MKLML_INNER_INCLUDE_DIR ${MKLML_INNER_INCLUDE_DIR} DIRECTORY)
list(APPEND IDEEP_INCLUDE_DIR ${MKLML_INNER_INCLUDE_DIR})
list(APPEND __ideep_looked_for ${MKLML_INNER_INCLUDE_DIR})
list(APPEND __ideep_looked_for MKLML_INNER_INCLUDE_DIR)
if(APPLE)
set(__mklml_inner_libs mklml iomp5)