diff options
Diffstat (limited to 'boltzgen/kernel/template/equilibrilize.cl.mako')
| -rw-r--r-- | boltzgen/kernel/template/equilibrilize.cl.mako | 15 |
1 files changed, 12 insertions, 3 deletions
diff --git a/boltzgen/kernel/template/equilibrilize.cl.mako b/boltzgen/kernel/template/equilibrilize.cl.mako index 0759dd5..4b9b984 100644 --- a/boltzgen/kernel/template/equilibrilize.cl.mako +++ b/boltzgen/kernel/template/equilibrilize.cl.mako @@ -1,6 +1,6 @@ -__kernel void equilibrilize(__global ${float_type}* f_next, - __global ${float_type}* f_prev, - unsigned int gid) +__kernel void equilibrilize_gid(__global ${float_type}* f_next, + __global ${float_type}* f_prev, + unsigned int gid) { __global ${float_type}* preshifted_f_next = f_next + gid; __global ${float_type}* preshifted_f_prev = f_prev + gid; @@ -10,3 +10,12 @@ __kernel void equilibrilize(__global ${float_type}* f_next, preshifted_f_prev[${layout.pop_offset(i)}] = ${w_i}.f; % endfor } + +% if 'cell_list_dispatch' in extras: +__kernel void equilibrilize_cells(__global ${float_type}* f_next, + __global ${float_type}* f_prev, + __global unsigned int* cells) +{ + equilibrilize_gid(f_next, f_prev, cells[get_global_id(0)]); +} +% endif |