From fd886e77eb2425337f7c010400ee9e5f996e48cd Mon Sep 17 00:00:00 2001
From: Maximilian Gaul <ptfs410h@mailhub.rrze.uni-erlangen.de>
Date: Mon, 8 Nov 2021 20:32:12 +0100
Subject: [PATCH] Added make config for NVCC

---
 config.mk       |  4 ++--
 include_NVCC.mk | 15 +++++++++++++++
 src/force.c     |  2 ++
 3 files changed, 19 insertions(+), 2 deletions(-)
 create mode 100644 include_NVCC.mk

diff --git a/config.mk b/config.mk
index 0765626..49612ce 100644
--- a/config.mk
+++ b/config.mk
@@ -1,5 +1,5 @@
-# Compiler tag (GCC/CLANG/ICC)
-TAG ?= CLANG
+# Compiler tag (GCC/CLANG/ICC/NVCC)
+TAG ?= NVCC
 # Enable likwid (true or false)
 ENABLE_LIKWID ?= false
 # SP or DP
diff --git a/include_NVCC.mk b/include_NVCC.mk
new file mode 100644
index 0000000..6571fe1
--- /dev/null
+++ b/include_NVCC.mk
@@ -0,0 +1,15 @@
+CC  = nvcc
+LINKER = $(CC)
+
+ANSI_CFLAGS  = -ansi
+ANSI_CFLAGS += -std=c99
+ANSI_CFLAGS += -pedantic
+ANSI_CFLAGS += -Wextra
+
+# CFLAGS   = -O0 -g  -std=c99 -fargument-noalias
+CFLAGS   = -O3 -arch=sm_61 # -fopenmp
+ASFLAGS  =  -masm=intel
+LFLAGS   =
+DEFINES  = -D_GNU_SOURCE -DLIKWID_PERFMON
+INCLUDES = $(LIKWID_INC)
+LIBS     = -lm $(LIKWID_LIB) -llikwid
diff --git a/src/force.c b/src/force.c
index 5ec92e7..f16f865 100644
--- a/src/force.c
+++ b/src/force.c
@@ -20,6 +20,8 @@
  *   with MD-Bench.  If not, see <https://www.gnu.org/licenses/>.
  * =======================================================================================
  */
+#include <cuda_runtime.h>
+
 #include <likwid-marker.h>
 
 #include <timing.h>