From cdea91abc66436f1ab317f3414177b577612371e Mon Sep 17 00:00:00 2001
From: James Price <j.price@bristol.ac.uk>
Date: Thu, 23 Feb 2017 17:49:46 +0000
Subject: [PATCH] Add required ARCH flag for RAJA on GPUs

---
 RAJA.make | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/RAJA.make b/RAJA.make
index 0f9cd61..5851e4f 100644
--- a/RAJA.make
+++ b/RAJA.make
@@ -24,7 +24,11 @@ endif
 
 else ifeq ($(TARGET), GPU)
 COMP = nvcc
-CFLAGS = --expt-extended-lambda -O3 -std=c++11 -x cu -Xcompiler -fopenmp
+
+ifndef ARCH
+$(error No ARCH defined. Specify target GPU architecture (e.g. ARCH=sm_35))
+endif
+CFLAGS = --expt-extended-lambda -O3 -std=c++11 -x cu -Xcompiler -fopenmp -arch $(ARCH)
 endif
 
 raja-stream: main.cpp RAJAStream.cpp