add make lib_cuda

Author: nvidia

MANIFEST.in

@@ -0,0 +1,5 @@
+include src/**
+include src/**/**
+include src/**/**/**
+include include/**
+include Makefile

Makefile

@@ -11,13 +11,14 @@ out:
 	mkdir out
 
 test:
-	# ${PYTHON_BIN} tests/test.py
-	${PYTHON_BIN} tests/test-with-multiprocessing.py
+	${PYTHON_BIN} tests/test.py
+	# ${PYTHON_BIN} tests/test-with-multiprocessing.py
 
 pynvjpeg: build/lib/libcolor_space.a
-	# rm build/lib.linux-aarch64-3.6/nvjpeg.cpython-36m-aarch64-linux-gnu.so -f
 	${PYTHON_BIN} setup.py build
 
+lib_cuda: build/lib/libcolor_space.a
+
 build/lib/libcolor_space.a: src/jetson/Utils/ColorSpace.cu
 	mkdir -p build/lib
 	nvcc -DCUDNN  --compiler-options "-fPIC -lstdc++ -pthread -lm" -c src/jetson/Utils/ColorSpace.cu -o build/lib/libcolor_space.a

setup.py

@@ -14,6 +14,7 @@
 if platform.system() == 'Linux':
     if os.path.exists('/usr/src/jetson_multimedia_api'):
         # Jetson
+        os.system('make lib_cuda')
         extension_nvjpeg = Extension('nvjpeg', 
             [
                 'nvjpeg-python.cpp', 'src/jetson/JpegCoder.cpp',