diff --git a/setup.py b/setup.py index 95ac9cc8fb16fb77caa4a8eea1eb8568aeaec607..1694fd308ed383f4a24dabc4f5e6e157ff29f959 100644 --- a/setup.py +++ b/setup.py @@ -12,11 +12,22 @@ def get_config_schema(): if 'darwin' in sys.platform: import platform osx_ver, _, _ = platform.mac_ver() - osx_ver = float('.'.join(osx_ver.split('.')[:2])) + osx_ver = '.'.join(osx_ver.split('.')[:2]) + + sysroot_paths = [ + "/Applications/Xcode.app/Contents/Developer/Platforms/MacOSX.platform/Developer/SDKs/MacOSX%s.sdk" % osx_ver, + "/Developer/SDKs/MacOSX%s.sdk" % osx_ver + ] default_libs = [] - default_cxxflags = ['-arch', 'i386', '-arch', 'x86_64', - '-isysroot', '/Developer/SDKs/MacOSX%s.sdk' % osx_ver] + default_cxxflags = ['-arch', 'i386', '-arch', 'x86_64'] + + from os.path import isdir + for srp in sysroot_paths: + if isdir(srp): + default_cxxflags.extend(['-isysroot', srp]) + break + default_ldflags = default_cxxflags[:] + ["-Wl,-framework,OpenCL"] else: