Added NVTX profile ranges through toggleable macros

CMakeLists.txt

@@ -64,6 +64,7 @@ set( with-includes OFF CACHE STRING "Add additional include paths [default=OFF].
 set( with-defines OFF CACHE STRING "Additional defines, e.g. '-DXYZ=1' [default=OFF]. Separate multiple defines by ';'." )
 set( with-max-rreg-count "55" CACHE STRING "Set a maximum amount of register used when compiling [default=55]. Separate multiple flags by ';'." )
 set( with-ptxas-options OFF CACHE STRING "Options for ptxas compiling [default=OFF]. Separate multiple flags by ';'." )
+set( with-nvtx OFF CACHE STRING "Enable compilation with NVTX ranges." )
 
 # generic build configuration
 set( with-version-suffix OFF CACHE STRING "Set a user defined version suffix [default='']." )
@@ -130,6 +131,7 @@ nest_process_with_optimize()
 nest_process_with_debug()
 nest_process_with_warning()
 nestgpu_post_process_compile_flags()
+nestgpu_process_with_nvtx()
 
 nest_process_version_suffix()
 

cmake/ProcessOptions.cmake

@@ -209,6 +209,14 @@ function( NESTGPU_POST_PROCESS_COMPILE_FLAGS )
 endfunction()
 
 
+function( NESTGPU_PROCESS_WITH_NVTX )
+  set( HAVE_NVTX OFF PARENT_SCOPE )
+  if ( with-nvtx )
+    set( HAVE_NVTX ON PARENT_SCOPE )
+  endif()
+endfunction()
+
+
 function( NEST_PROCESS_VERSION_SUFFIX )
   if ( with-version-suffix )
     foreach ( flag ${with-version-suffix} )

libnestutil/config.h.in

@@ -48,6 +48,9 @@
 /* Define to 1 if you have the <unistd.h> header file. */
 #cmakedefine HAVE_UNISTD_H 1
 
+/* Define to 1 for compiling with nvtx ranges. */
+#cmakedefine HAVE_NVTX 1
+
 /* Version number of package */
 #define VERSION @NEST_GPU_VERSION_STRING@
 

src/CMakeLists.txt

@@ -75,6 +75,7 @@ nestgpu_C.h
 neuron_models.h
 ngpu_exception.h
 node_group.h
+nvtx_macros.h
 parrot_neuron.h
 poiss_gen.h
 poiss_gen_variables.h