From bc96bb3d76bd28891d079423f35b0c6a90a5b6bd Mon Sep 17 00:00:00 2001
From: Bert Peters <bert@bertptrs.nl>
Date: Mon, 5 Mar 2018 16:23:28 +0100
Subject: [PATCH] Clean up CMakeLists.

---
 CMakeLists.txt | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 43029d3..ebf0983 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -7,8 +7,8 @@ set(CMAKE_MODULE_PATH ${CMAKE_MODULE_PATH} "${CMAKE_SOURCE_DIR}/cmake/modules")
 # Enable modern C++ features
 set(CMAKE_CXX_STANDARD 17)
 
-# More warnings
-set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wall -Wextra -pedantic")
+# Enable better warnings
+set(COMPILE_OPTIONS "-Wall -Wextra -pedantic")
 
 # Define executable and dependencies
 file(GLOB fmri_SRC
@@ -18,8 +18,14 @@ file(GLOB fmri_SRC
 
 add_executable(fmri ${fmri_SRC})
 
-# Build without GPU support for quicker development
-add_definitions(-DCPU_ONLY)
+# Enable CUDA if available
+find_package(CUDA)
+if (CUDA_FOUND)
+	include_directories(${CUDA_INCLUDE_DIRS})
+	target_link_libraries(fmri ${CUDA_LIBS})
+else(CUDA_FOUND)
+	add_definitions(-DCPU_ONLY)
+endif(CUDA_FOUND)
 
 
 # Library dependencies