Updates and some refactoring

utils/CMakeLists.txt

@@ -1,7 +1,7 @@
 project(PointNet2)
 cmake_minimum_required(VERSION 3.5)
 
-find_package(CUDA)
+find_package(CUDA REQUIRED)
 
 include_directories("${CMAKE_SOURCE_DIR}/cinclude")
 cuda_include_directories("${CMAKE_SOURCE_DIR}/cinclude")
@@ -9,11 +9,13 @@ file(GLOB cuda_kernels_src "csrc/*.cu")
 cuda_compile(cuda_kernels SHARED ${cuda_kernels_src} OPTIONS -O3)
 
 file(GLOB wrapper_headers "cinclude/*wrapper.h")
+file(GLOB wrapper_sources "csrs/*.c")
 add_custom_command(OUTPUT "${CMAKE_SOURCE_DIR}/_ext/__ext.so"
 		   WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}
 		   COMMAND python "${CMAKE_SOURCE_DIR}/build_ffi.py" ${cuda_kernels}
 		   DEPENDS ${cuda_kernels}
 		   DEPENDS ${wrapper_headers}
+		   DEPENDS ${wrapper_sources}
 		   VERBATIM)
 
 add_custom_target(ext ALL