/usr/lib/python2.7/dist-packages/pyopencl/compyte/ndarray/setup_opencl.py is in python-pyopencl 2017.2.2-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 | from __future__ import absolute_import
import os
from distutils.core import setup, Extension
from distutils.command.build_ext import build_ext
from distutils.dep_util import newer
import numpy as np
class build_ext_nvcc(build_ext):
user_options = build_ext.user_options
user_options.extend([
('cuda-root=', None, "The cuda root directory")])
def initialize_options(self):
build_ext.initialize_options(self)
self.cuda_root = None
def finalize_options(self):
build_ext.finalize_options(self)
if self.cuda_root is None:
self.cuda_root = os.getenv('CUDA_ROOT', None)
if self.cuda_root is not None:
self._nvcc_bin = os.path.join(self.cuda_root, 'bin', 'nvcc')
else:
self._nvcc_bin = 'nvcc'
def cuda_process(self, source, include_args):
target = source + '.cpp'
if newer(source, target):
self.spawn([self._nvcc_bin, '--cuda', source, '-o', target] + \
include_args)
return target
def cuda_extension(self, ext):
includes = self.distribution.include_dirs + ext.include_dirs
include_args = ['-I' + i for i in includes]
new_sources = []
anycuda = False
for src in ext.sources:
if src.endswith('.cu'):
new_sources.append(self.cuda_process(src, include_args))
anycuda = True
else:
new_sources.append(src)
if anycuda:
ext.sources = new_sources
if self.cuda_root is not None:
lib = os.path.join(self.cuda_root, 'lib')
lib64 = os.path.join(self.cuda_root, 'lib64')
if os.path.isdir(lib):
ext.library_dirs.append(lib)
ext.extra_link_args.append('-Xlinker')
ext.extra_link_args.append('-rpath')
ext.extra_link_args.append('-Xlinker')
ext.extra_link_args.append(lib)
if os.path.isdir(lib64):
ext.library_dirs.append(lib64)
# ext.extra_link_args.append('-rpath')
# ext.extra_link_args.append(lib64)
if 'cudart' not in ext.libraries:
ext.libraries.append('cudart')
if self.cuda_root:
include = os.path.join(self.cuda_root, 'include')
if os.path.isdir(include):
ext.extra_compile_args.append('-I' + include)
if os.path.isfile('/usr/lib/nvidia-current/libOpenCL.so'):
ext.extra_link_args.append('-L/usr/lib/nvidia-current')
ext.extra_link_args.append('-Xlinker')
ext.extra_link_args.append('-rpath')
ext.extra_link_args.append('-Xlinker')
ext.extra_link_args.append('/usr/lib/nvidia-current')
def build_extensions(self):
self.check_extensions_list(self.extensions)
for ext in self.extensions:
self.cuda_extension(ext)
# uncomment this + inherit from the cython version of build_ext
# work with cuda and cython sources
#ext.sources = self.cython_sources(ext.sources, ext)
self.build_extension(ext)
import sys
if sys.platform == 'darwin':
libcl_args = {'extra_link_args': ['-framework', 'OpenCL']}
else:
libcl_args = {'libraries': ['OpenCL']}
setup(name='compyte',
cmdclass={'build_ext': build_ext_nvcc},
include_dirs=[np.get_include(), '.'],
ext_modules=[Extension('pygpu_ndarray',
define_macros=[('OFFSET', '1'), ('WITH_OPENCL', '')],
sources=['pygpu_language_opencl.cpp',
'pygpu_ndarray.cpp'],
**libcl_args)
]
)
|