Merge pull request #27865 from perfinion:sysconfig

PiperOrigin-RevId: 243664878
This commit is contained in:
TensorFlower Gardener 2019-04-15 12:48:38 -07:00
commit e9e1d82e55

View File

@ -19,6 +19,7 @@ from __future__ import division
from __future__ import print_function
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 MONOLITHIC_BUILD as _MONOLITHIC_BUILD
@ -73,8 +74,13 @@ def get_link_flags():
Returns:
The link flags.
"""
is_mac = _platform.system() == 'Darwin'
ver = _VERSION.split('.')[0]
flags = []
if not _MONOLITHIC_BUILD:
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