nms::setup_linux Namespace Reference

Classes
class	custom_build_ext
Functions
def	customize_compiler_for_nvcc
def	find_in_path
def	locate_cuda
Variables
	_compile
dictionary	cmdclass = {'build_ext': custom_build_ext}
	compiler_so
tuple	CUDA = locate_cuda()
list	ext_modules
string	name = 'nms'
tuple	numpy_include = np.get_include()

---

Function Documentation

def nms.setup_linux.customize_compiler_for_nvcc

(

self

)

inject deep into distutils to customize how the dispatch
to gcc/nvcc works.
If you subclass UnixCCompiler, it's not trivial to get your subclass
injected in, and still have the right customizations (i.e.
distutils.sysconfig.customize_compiler) run on it. So instead of going
the OO route, I have this. Note, it's kindof like a wierd functional
subclassing going on.

Definition at line 66 of file nms/setup_linux.py.

def nms.setup_linux.find_in_path	(		name,
			path
	)

Definition at line 16 of file nms/setup_linux.py.

def nms.setup_linux.locate_cuda

(

)

Locate the CUDA environment on the system
Returns a dict with keys 'home', 'nvcc', 'include', and 'lib64'
and values giving the absolute path to each directory.
Starts by looking for the CUDAHOME env variable. If not found, everything
is based on finding 'nvcc' in the PATH.

Definition at line 27 of file nms/setup_linux.py.

---

Variable Documentation

nms::setup_linux::_compile

Definition at line 72 of file nms/setup_linux.py.

dictionary nms::setup_linux::cmdclass = {'build_ext': custom_build_ext}

Definition at line 140 of file nms/setup_linux.py.

nms::setup_linux::compiler_so

Definition at line 72 of file nms/setup_linux.py.

tuple nms::setup_linux::CUDA = locate_cuda()

Definition at line 56 of file nms/setup_linux.py.

nms::setup_linux::ext_modules

Initial value:

[
    Extension(
        "cpu_nms",
        ["cpu_nms.pyx"],
        extra_compile_args={'gcc': ["-Wno-cpp", "-Wno-unused-function"]},
        include_dirs = [numpy_include]
    ),
    Extension('gpu_nms',
        ['nms_kernel.cu', 'gpu_nms.pyx'],
        library_dirs=[CUDA['lib64']],
        libraries=['cudart'],
        language='c++',
        runtime_library_dirs=[CUDA['lib64']],
        # this syntax is specific to this build system
        # we're only going to use certain compiler args with nvcc and not with
        # gcc the implementation of this trick is in customize_compiler() below
        extra_compile_args={'gcc': ["-Wno-unused-function"],
                            'nvcc': ['-arch=sm_35',
                                     '--ptxas-options=-v',
                                     '-c',
                                     '--compiler-options',
                                     "'-fPIC'"]},
        include_dirs = [numpy_include, CUDA['include']]
    ),
]

Definition at line 110 of file nms/setup_linux.py.

string nms::setup_linux::name = 'nms'

Definition at line 137 of file nms/setup_linux.py.

tuple nms::setup_linux::numpy_include = np.get_include()

Definition at line 61 of file nms/setup_linux.py.