frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-11-04 16:04:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

build_env.py: filter non-existent dirs from INCLUDES, LIBDIRS

Browse Source
This commit is contained in:
Jeff Donahue
2015-07-26 21:27:04 -07:00
parent 2d30d89cb1
commit a139a1bf6a
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
build_env.py
View File

@ -153,6 +153,7 @@ class Env(object):
'/usr/local/include',
'/usr/include',
]
INCLUDES = [s for s in INCLUDES if s == GENDIR or os.path.isdir(s)]
INCLUDES = ' '.join(['-I' + s for s in INCLUDES])
# Python
INCLUDES += ' ' + _GetPythonIncludes()
@ -162,6 +163,7 @@ class Env(object):
'/usr/local/lib',
'/usr/lib',
]
LIBDIRS = [s for s in LIBDIRS if os.path.isdir(s)]
LIBDIRS = ' '.join(['-L' + s for s in LIBDIRS])
# Python
LIBDIRS += ' ' + _GetPythonLibDirs()