Simplified uniform GPU selection in CMake

[alexxy/gromacs.git] / src / gromacs / gpu_utils / devicebuffer.h

diff --git a/src/gromacs/gpu_utils/devicebuffer.h b/src/gromacs/gpu_utils/devicebuffer.h
index c0cdfec32963b3a058477439fe9e1a7700349183..8aaea7226320564886844aeca3167cd87a26cd66 100644 (file)
--- a/src/gromacs/gpu_utils/devicebuffer.h
+++ b/src/gromacs/gpu_utils/devicebuffer.h
@@ -50,9 +50,9 @@
 #include "gromacs/utility/gmxassert.h"
 #include "gromacs/utility/smalloc.h" // TODO: this is only for over_alloc_large
 
-#if GMX_GPU == GMX_GPU_CUDA
+#if GMX_GPU_CUDA
 #    include "gromacs/gpu_utils/devicebuffer.cuh"
-#elif GMX_GPU == GMX_GPU_OPENCL
+#elif GMX_GPU_OPENCL
 #    include "gromacs/gpu_utils/devicebuffer_ocl.h"
 #else
 #    error "devicebuffer.h included on non-GPU build!"