From 2075b177ef05a300f067a08e999caa06d938e115 Mon Sep 17 00:00:00 2001
From: paboyle <paboyle@ph.ed.ac.uk>
Date: Wed, 13 Jun 2018 20:22:34 +0100
Subject: [PATCH] CUDA_ARCH more carefule treatment

---
 lib/Grid_Eigen_Dense.h | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/lib/Grid_Eigen_Dense.h b/lib/Grid_Eigen_Dense.h
index 05a3d791..dcbfa40b 100644
--- a/lib/Grid_Eigen_Dense.h
+++ b/lib/Grid_Eigen_Dense.h
@@ -10,8 +10,11 @@
 #pragma diag_suppress code_is_unreachable
 #undef __NVCC__
 #undef __CUDACC__
+#ifdef __CUDA_ARCH__
+#define __CUDA_ARCH__REDEFINE__
 #undef __CUDA_ARCH__
 #endif
+#endif
 
 #include <Grid/Eigen/Dense>
 
@@ -19,8 +22,11 @@
 #pragma pop
 #define __NVCC__
 #define __CUDACC__
+
+#ifdef __CUDA_ARCH__REDEFINE__
 #define __CUDA_ARCH__
 #endif
+#endif
 
 #if defined __GNUC__
 #pragma GCC diagnostic pop