diff --git a/ocl-stream-kernels.cl b/ocl-stream-kernels.cl
index 762af85..eeb47f4 100644
--- a/ocl-stream-kernels.cl
+++ b/ocl-stream-kernels.cl
@@ -1,7 +1,11 @@
 
-#pragma OPENCL EXTENSION cl_khr_fp64 : enable
+#ifdef FLOAT
+	#define DATATYPE float
+#else
+	#pragma OPENCL EXTENSION cl_khr_fp64 : enable
+	#define DATATYPE double
+#endif
 
-#define DATATYPE double
 
 constant DATATYPE scalar = 3.0;