From 6480dc29fbca7ad03a6834ebd469ea948e348138 Mon Sep 17 00:00:00 2001 From: pascal Date: Sat, 17 Nov 2012 02:52:35 +0100 Subject: [PATCH] add lib-subfolders for Linux AMDAPP newer versions of AMDAPP seem to use not lib/*.so but architecture-specific subfolders lib/x86 or lib/x86_64 --- setup.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 3c5494e..e0acc82 100644 --- a/setup.py +++ b/setup.py @@ -41,8 +41,9 @@ else: AMDAPPSDKROOT = os.environ.get('AMDAPPSDKROOT', '/usr/local') - - flags = dict(libraries=['OpenCL'], include_dirs=[join(AMDAPPSDKROOT, 'include')], library_dirs=[join(AMDAPPSDKROOT, 'lib')]) + def amd_dir(dirs): + return map(lambda x: join(AMDAPPSDKROOT, x), dirs) + flags = dict(libraries=['OpenCL'], include_dirs=amd_dir(['include']), library_dirs=amd_dir(['lib', 'lib/x86', 'lib/x86_64'])) extension = lambda name, ext: Extension('.'.join(('opencl', name)), [join('opencl', name + ext)], **flags) pyx_extention_names = [name[:-4] for name in os.listdir('opencl') if name.endswith('.pyx')]