diff --git a/heavy/heavy.cu b/heavy/heavy.cu index 812a872..936006a 100644 --- a/heavy/heavy.cu +++ b/heavy/heavy.cu @@ -1,9 +1,15 @@ #include #include +#include #include -// include thrust -#define USE_TRUST 1 +// include thrust if possible +#if defined(__GNUC__) && __GNUC__ == 5 && __GNUC_MINOR__ >= 2 && CUDA_VERSION < 7000 +#warning "Heavy: incompatible GCC version!" +#define USE_THRUST 0 +#else +#define USE_THRUST 1 +#endif #if USE_THRUST #include