aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--template/kernel.mako8
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 {