Opt signal handling, LambdaGPUTask (#9)

* Optional signal handling

* The next level GPU tasks with lambda functions!

* GPU calculations take a parameter

CMakeLists.txt

@@ -12,6 +12,7 @@ set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -Werror cross-execution-space-call,reo
 
 set(CMAKE_CXX_STANDARD 11)
 set(CMAKE_CXX_FLAGS "-O2 -Wall -Wextra -Wshadow -Wpedantic -Wnon-virtual-dtor -Wunused -Wdouble-promotion")
+set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -g -rdynamic")
 set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Werror")
 
 

include/cuq.h

@@ -1,5 +1,6 @@
 #include <mutex>
 #include <queue>
+#include <functional>
 
 #pragma once
 
@@ -9,6 +10,23 @@ class GPUTask {
     virtual ~GPUTask();
 };
 
+template <class T>
+class LambdaGPUTask : public GPUTask {
+  public:
+    LambdaGPUTask(T _params, std::function<void(T)> _gpuCalculations) {
+      params = _params;
+      gpuCalculations = _gpuCalculations;
+    }
+
+    void doWork() {
+      gpuCalculations(params);
+    }
+
+  private:
+    T params;
+    std::function<void(T)> gpuCalculations;
+};
+
 class GPUTasksQueue {
   public:
     bool resetDeviceAfterFinish;
@@ -27,7 +45,7 @@ extern "C"
 void processTasks(
   GPUTask ** tasks, int taskCount,
   int requestedDevicesCount, 
-  bool resetDeviceAfterFinish = false, bool deleteTasksAutomatically = true);
+  bool resetDeviceAfterFinish = false, bool deleteTasksAutomatically = true, bool handleSignals = true);
 
 extern "C"
 int occupyDevices(int requestedDevicesCount, int * occupiedDevicesIdxs, char * errorMsg);

src/cuq.cpp

@@ -104,9 +104,10 @@ extern "C"
 void processTasks(
   GPUTask ** tasks, int taskCount,
   int requestedDevicesCount, 
-  bool resetDeviceAfterFinish, bool deleteTasksAutomatically) {
+  bool resetDeviceAfterFinish, bool deleteTasksAutomatically, bool handleSignals) {
 
-  signal(SIGSEGV, signalHandler);
+  if (handleSignals)
+    signal(SIGSEGV, signalHandler);
 
   char errorMsg[1000];
   int devices[128];