diff options
Diffstat (limited to 'template')
| -rw-r--r-- | template/kernel.mako | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/template/kernel.mako b/template/kernel.mako index 6f82b20..24a6f64 100644 --- a/template/kernel.mako +++ b/template/kernel.mako @@ -7,7 +7,7 @@ __kernel void equilibrilize(__global __write_only float* f_a, __global __write_only float* preshifted_f_b = f_b + gid; % if pop_eq_src == '': -% for i, w_i in enumerate(w): +% for i, w_i in enumerate(descriptor.w): preshifted_f_a[${i*nCells}] = ${w_i}.f; preshifted_f_b[${i*nCells}] = ${w_i}.f; % endfor @@ -42,7 +42,7 @@ __kernel void collide_and_stream(__global __write_only float* f_a, __global __write_only float* preshifted_f_a = f_a + gid; __global __read_only float* preshifted_f_b = f_b + gid; -% for i, c_i in enumerate(c): +% for i, c_i in enumerate(descriptor.c): const float f_curr_${i} = preshifted_f_b[${direction_index(c_i)*nCells + neighbor_offset(-c_i)}]; % endfor @@ -64,7 +64,7 @@ __kernel void collide_and_stream(__global __write_only float* f_a, const float ${ccode(expr)} % endfor -% for i in range(0,len(c)): +% for i in range(0,descriptor.q): preshifted_f_a[${i*nCells}] = f_next_${i}; % endfor } @@ -76,7 +76,7 @@ __kernel void collect_moments(__global __read_only float* f, __global __read_only float* preshifted_f = f + gid; -% for i in range(0,len(c)): +% for i in range(0,descriptor.q): const float f_curr_${i} = preshifted_f[${i*nCells}]; % endfor |