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')]