Explicitly enable double precision floating point when required

1 files changed, 8 insertions, 0 deletions

diff --git a/template/kernel.mako b/template/kernel.mako
index bf7b29b..dd8eaee 100644
--- a/template/kernel.mako
+++ b/template/kernel.mako
@@ -1,3 +1,11 @@
+% if float_type == 'double':
+#if defined(cl_khr_fp64)
+#pragma OPENCL EXTENSION cl_khr_fp64 : enable
+#elif defined(cl_amd_fp64)
+#pragma OPENCL EXTENSION cl_amd_fp64 : enable
+#endif
+% endif
+
 <%
 def gid():
     return {