Corrected CMakeLists.txt

src/main/java/de/monticore/lang/monticar/cnnarch/caffe2generator/CNNArch2Caffe2.java

@@ -156,7 +156,14 @@ public class CNNArch2Caffe2 implements CNNArchGenerator{
 
         CMakeConfig cMakeConfig = new CMakeConfig(rootModelName);
         cMakeConfig.addModuleDependency(new CMakeFindModule("Armadillo", true));
-        cMakeConfig.addCMakeCommand("set(LIBS ${LIBS} mxnet)");
+        cMakeConfig.addModuleDependency(new CMakeFindModule("Caffe2", true));
+        cMakeConfig.addCMakeCommand("find_package(CUDA)" + "\n"
+                                        + "set(INCLUDE_DIRS ${INCLUDE_DIRS} ${CUDA_INCLUDE_DIRS})" + "\n"
+                                        + "set(LIBS ${LIBS} ${CUDA_LIBRARIES} ${CUDA_curand_LIBRARY})" + "\n"); //Needed since CUDA cannot be found correctly (including CUDA_curand_LIBRARY) and as optional using CMakeFindModule
+
+
+        cMakeConfig.addCMakeCommand("if(CUDA_FOUND)" + "\n" + "  set(LIBS ${LIBS} caffe2 caffe2_gpu)"
+                                        + "\n" + "else()" + "\n" + "  set(LIBS ${LIBS} caffe2)" + "\n" + "endif()");
 
         Map<String,String> fileContentMap = new HashMap<>();
         for (FileContent fileContent : cMakeConfig.generateCMakeFiles()){

src/test/resources/target_code/CMakeLists.txt

@@ -7,12 +7,23 @@ project(alexnet LANGUAGES CXX)
 set(CMAKE_MODULE_PATH ${CMAKE_MODULE_PATH} ${CMAKE_CURRENT_SOURCE_DIR}/cmake)
 
 # add dependencies
+find_package(Caffe2 REQUIRED)
+set(INCLUDE_DIRS ${INCLUDE_DIRS} ${Caffe2_INCLUDE_DIRS})
+set(LIBS ${LIBS} ${Caffe2_LIBRARIES})
 find_package(Armadillo REQUIRED)
 set(INCLUDE_DIRS ${INCLUDE_DIRS} ${Armadillo_INCLUDE_DIRS})
 set(LIBS ${LIBS} ${Armadillo_LIBRARIES})
 
 # additional commands
-set(LIBS ${LIBS} mxnet)
+find_package(CUDA)
+set(INCLUDE_DIRS ${INCLUDE_DIRS} ${CUDA_INCLUDE_DIRS})
+set(LIBS ${LIBS} ${CUDA_LIBRARIES} ${CUDA_curand_LIBRARY})
+
+if(CUDA_FOUND)
+  set(LIBS ${LIBS} caffe2 caffe2_gpu)
+else()
+  set(LIBS ${LIBS} caffe2)
+endif()
 
 # create static library
 include_directories(${INCLUDE_DIRS})