Fix MacOS build

Signed-off-by: caozhong <[email protected]>

Author: CaoZhongZ

CMakeLists.txt

@@ -10,6 +10,11 @@ set(IDEEP_VERSION
 
 option(multinode "Provide non-blocking communciation support for machine learning" OFF)
 
+IF(APPLE)
+  SET(CMAKE_INSTALL_NAME_DIR @rpath)
+  SET(CMAKE_BUILD_WITH_INSTALL_RPATH ON)
+ENDIF(APPLE)
+
 set(USE_MKLDNN_INTERNAL 1)
 include(cmake/profiling.cmake)
 include(cmake/mkldnn.cmake)

python/setup.py

@@ -145,12 +145,11 @@ def run(self):
             ]
 
 if os_name == 'Linux':
-    libraries = ['mkldnn', 'mklml_intel', 'ideep']  # , 'dlcomp']
+    libraries = ['mkldnn', 'mklml_intel', 'ideep', 'm']  # , 'dlcomp']
     ccxx_opts += ['-fopenmp']
-    libraries += ['m']
     link_opts += ['-Wl,-z,now', '-Wl,-z,noexecstack']
 else:
-    libraries = ['mkldnn', 'mklml']
+    libraries = ['mkldnn', 'mklml', 'ideep']
 
 src = ['ideep4py/py/ideep4py.i',
        'ideep4py/py/mm/mdarray.cc',