diff options
Diffstat (limited to 'boltzgen/kernel/template/pattern/AB.cl.mako')
| -rw-r--r-- | boltzgen/kernel/template/pattern/AB.cl.mako | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/boltzgen/kernel/template/pattern/AB.cl.mako b/boltzgen/kernel/template/pattern/AB.cl.mako index d2d8b8a..33836ea 100644 --- a/boltzgen/kernel/template/pattern/AB.cl.mako +++ b/boltzgen/kernel/template/pattern/AB.cl.mako @@ -66,4 +66,26 @@ __kernel void ${name}( ${caller.body()} } + +% if 'cell_list_dispatch' in extras: +__kernel void ${name}_cells( + __global ${float_type}* f + , __global unsigned int* cells +% if params is not None: +% for param_type, param_name in params: + , ${param_type} ${param_name} +% endfor +% endif +) { + ${name}( + f + , cells[get_global_id(0)] +% if params is not None: +% for param_type, param_name in params: + , ${param_name} +% endfor +% endif + ); +} +% endif </%def> |