diff options
-rw-r--r-- | template/kernel.mako | 8 |
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 { |