Merge pull request #123 from nlesc-recruit/116-vector-add

CMakeLists.txt

@@ -17,10 +17,14 @@ if(NOT EXISTS ${PROJECT_BINARY_DIR}/.gitignore)
 endif()
 
 set(SOURCE_FILES src/cu.cpp src/nvrtc.cpp)
+
 include_directories(./include/)
 include_directories(${CUDAToolkit_INCLUDE_DIRS})
 
 add_library(cudawrappers SHARED ${SOURCE_FILES})
 
 # Including linters rules
 include(cmake/linter-tools.cmake)
+
+# Tests
+add_subdirectory(tests)

README.dev.md

@@ -241,7 +241,16 @@ where `<tool>` can be any of the following:
 
 ## Running the tests
 
-:construction:
+Enter the `build` directory and run `make test`.
+
+If you are running the tests on DAS, you can run a job using `srun` command.
+For instance,
+
+```shell
+srun -N 1 -C TitanX --gres=gpu:1 make test
+```
+
+This command will run the tests in one of the worker nodes with a GPU device.
 
 ## Building the API documentation
 

tests/CMakeLists.txt

@@ -0,0 +1,8 @@
+# vector_add
+add_subdirectory(vector_add)
+
+# This command will be executed from the test folder, which is where the .cu file must be located
+add_custom_target(test
+  COMMAND vector_add/vector_add
+  DEPENDS vector_add
+)

tests/vector_add/CMakeLists.txt

@@ -0,0 +1,5 @@
+# Copies the kernel function to the tests folder
+file(COPY vector_add_kernel.cu DESTINATION ..)
+add_executable(vector_add EXCLUDE_FROM_ALL vector_add.cpp)
+target_link_libraries(vector_add PRIVATE cudawrappers CUDA::cuda_driver
+                                         CUDA::nvrtc)

tests/vector_add/vector_add.cpp

@@ -1,7 +1,7 @@
 #include <iostream>
 
-#include "cu/cu.h"
-#include "cu/nvrtc.h"
+#include "cu.hpp"
+#include "nvrtc.hpp"
 
 void vector_add() {
   int N = 1024;