diff --git a/Makefile b/Makefile index 8e42705..d098f91 100644 --- a/Makefile +++ b/Makefile @@ -20,7 +20,7 @@ OBJDIR=./obj/ CC=gcc AR=ar NVCC=nvcc -OPTS=-Ofast +OPTS=-Ofast -fopenmp -lm -lmvec COMMON= CFLAGS=-Wall -Wfatal-errors -Wno-unused-result -fPIC CFLAGS+=-I./darknet/src -I./darknet/include diff --git a/setup.py b/setup.py index 626dde4..5b45b36 100644 --- a/setup.py +++ b/setup.py @@ -4,6 +4,8 @@ module = Extension('pyyolo', library_dirs=['.'], libraries=['yolo'], + extra_compile_args=['-fopenmp'], + extra_link_args=['-lgomp', '-lm', '-lmvec'], include_dirs=[numpy.get_include(), './darknet/include'], sources = ['module.c']) diff --git a/setup_gpu.py b/setup_gpu.py index 64003b1..d694d6e 100644 --- a/setup_gpu.py +++ b/setup_gpu.py @@ -4,6 +4,8 @@ module = Extension('pyyolo', library_dirs=['.', '/usr/local/cuda/lib64', '/usr/local/'], libraries=['yolo', 'cuda', 'cudart', 'cublas', 'curand', 'cudnn'], + extra_compile_args=['-fopenmp'], + extra_link_args=['-lgomp', '-lm', '-lmvec'], include_dirs=[numpy.get_include(), './darknet/include'], sources = ['module.c'])