.

CMakeLists.txt

@@ -17,10 +17,19 @@ endif(INCLUDE_CUDA)
 set(SOURCES_CXX 
     src/MemoryProcessing.cpp
     )
+set(HEADERS_CXX 
+    include/MemoryProcessing.h
+    )
 if(INCLUDE_CUDA)
     set(SOURCES_CU 
     src/MemoryProcessing.cu
     )
+    set(HEADERS_CU
+    include/MemoryProcessing.cuh
+    )
 endif(INCLUDE_CUDA)
 
-add_library(memproc STATIC ${SOURCES_CU} ${SOURCES_CXX})
+set(SOURCES ${SOURCES_CU} ${SOURCES_CXX})
\ No newline at end of file
+set(HEADERS ${HEADERS_CU} ${HEADERS_CXX})
+
+add_library(memproc STATIC ${HEADERS} ${SOURCES})
\ No newline at end of file