experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

sysconfig: fix library name on macos

Browse Source 
On darwin, the library is named libtensorflow_framework.2.dylib

Signed-off-by: Jason Zaman <jason@perfinion.com>
This commit is contained in:
Jason Zaman 2019-04-16 01:22:23 +08:00
parent 79d8779069
commit e2994c09ec
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/python/platform/sysconfig.py
View File

@ -19,6 +19,7 @@ from __future__ import division
from __future__ import print_function from __future__ import print_function
import os.path as _os_path import os.path as _os_path
import platform as _platform
from tensorflow.python.framework.versions import CXX11_ABI_FLAG as _CXX11_ABI_FLAG from tensorflow.python.framework.versions import CXX11_ABI_FLAG as _CXX11_ABI_FLAG
from tensorflow.python.framework.versions import MONOLITHIC_BUILD as _MONOLITHIC_BUILD from tensorflow.python.framework.versions import MONOLITHIC_BUILD as _MONOLITHIC_BUILD
@ -73,8 +74,13 @@ def get_link_flags():
Returns: Returns:
The link flags. The link flags.
""" """
is_mac = _platform.system() == 'Darwin'
ver = _VERSION.split('.')[0]
flags = [] flags = []
if not _MONOLITHIC_BUILD: if not _MONOLITHIC_BUILD:
flags.append('-L%s' % get_lib()) flags.append('-L%s' % get_lib())
flags.append('-l:libtensorflow_framework.so.%s' % _VERSION.split('.')[0]) if is_mac:
flags.append('-l:libtensorflow_framework.%s.dylib' % ver)
else:
flags.append('-l:libtensorflow_framework.so.%s' % ver)
return flags return flags