nms::setup_windows Namespace Reference

Classes
class	custom_build_ext
Functions
def	customize_compiler_for_nvcc
def	find_in_path
def	locate_cuda
Variables
dictionary	cmdclass = {'build_ext': custom_build_ext}
	compile
tuple	CUDA = locate_cuda()
list	ext_modules
string	lib_dir = 'lib/x64'
string	name = 'fast_rcnn'
tuple	numpy_include = np.get_include()
string	nvcc_bin = 'nvcc.exe'

---

Function Documentation

def nms.setup_windows.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 80 of file nms/setup_windows.py.

def nms.setup_windows.find_in_path	(		name,
			path
	)

Definition at line 25 of file nms/setup_windows.py.

def nms.setup_windows.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 36 of file nms/setup_windows.py.

---

Variable Documentation

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

Definition at line 144 of file nms/setup_windows.py.

nms::setup_windows::compile

Definition at line 87 of file nms/setup_windows.py.

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

Definition at line 70 of file nms/setup_windows.py.

nms::setup_windows::ext_modules

Initial value:

[
    # unix _compile: obj, src, ext, cc_args, extra_postargs, pp_opts
    Extension(
        "cpu_nms",
        sources=["cpu_nms.pyx"],
        extra_compile_args={'gcc': []},
        include_dirs = [numpy_include],
    ),
]

Definition at line 130 of file nms/setup_windows.py.

string nms::setup_windows::lib_dir = 'lib/x64'

Definition at line 19 of file nms/setup_windows.py.

string nms::setup_windows::name = 'fast_rcnn'

Definition at line 141 of file nms/setup_windows.py.

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

Definition at line 75 of file nms/setup_windows.py.

string nms::setup_windows::nvcc_bin = 'nvcc.exe'

Definition at line 18 of file nms/setup_windows.py.